From cb9c26035906149a2fbf9117437ef76967c90231 Mon Sep 17 00:00:00 2001
From: litagin02 <litagin02@gmail.com>
Date: Sun, 4 Feb 2024 16:55:58 +0900
Subject: [PATCH] Fix is_resuming bug for JP-Extra

---
 utils.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/utils.py b/utils.py
index 33f6e10..ca194bf 100644
--- a/utils.py
+++ b/utils.py
@@ -189,10 +189,11 @@ def summarize(
 
 
 def is_resuming(dir_path):
+    # JP-ExtraバージョンではDURがなくWDがあったり変わるため、Gのみで判断する
     g_list = glob.glob(os.path.join(dir_path, "G_*.pth"))
-    d_list = glob.glob(os.path.join(dir_path, "D_*.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
+    # d_list = glob.glob(os.path.join(dir_path, "D_*.pth"))
+    # dur_list = glob.glob(os.path.join(dir_path, "DUR_*.pth"))
+    return len(g_list) > 0
 
 
 def latest_checkpoint_path(dir_path, regex="G_*.pth"):