tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Fix is_resuming bug for JP-Extra

Browse Source
This commit is contained in:
litagin02
2024-02-04 16:55:58 +09:00
parent 4714456dfe
commit cb9c260359
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
utils.py
View File

@@ -189,10 +189,11 @@ def summarize(
def is_resuming(dir_path): def is_resuming(dir_path):
# JP-ExtraバージョンではDURがなくWDがあったり変わるため、Gのみで判断する
g_list = glob.glob(os.path.join(dir_path, "G_*.pth")) g_list = glob.glob(os.path.join(dir_path, "G_*.pth"))
d_list = glob.glob(os.path.join(dir_path, "D_*.pth")) # d_list = glob.glob(os.path.join(dir_path, "D_*.pth"))
dur_list = glob.glob(os.path.join(dir_path, "DUR_*.pth")) # dur_list = glob.glob(os.path.join(dir_path, "DUR_*.pth"))
return len(g_list) > 0 and len(d_list) > 0 and len(dur_list) > 0 return len(g_list) > 0
def latest_checkpoint_path(dir_path, regex="G_*.pth"): def latest_checkpoint_path(dir_path, regex="G_*.pth"):