Update checkpoint filenames in modl example scripts

lanl · Nov 13, 2023 · 1bcc0d5 · 1bcc0d5
1 parent 6994a66
commit 1bcc0d5
Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/scripts/ct_astra_modl_train_foam2.py b/examples/scripts/ct_astra_modl_train_foam2.py
@@ -169,7 +169,7 @@
 
 checkpoint_files = []
 for dirpath, dirnames, filenames in os.walk(workdir2):
-    checkpoint_files = [fn for fn in filenames if str.split(fn, "_")[0] == "checkpoint"]
+    checkpoint_files = [fn for fn in filenames]
 
 if len(checkpoint_files) > 0:
     model = sflax.MoDLNet(

diff --git a/examples/scripts/deconv_modl_train_foam1.py b/examples/scripts/deconv_modl_train_foam1.py
@@ -164,7 +164,7 @@
 
 checkpoint_files = []
 for dirpath, dirnames, filenames in os.walk(workdir2):
-    checkpoint_files = [fn for fn in filenames if str.split(fn, "_")[0] == "checkpoint"]
+    checkpoint_files = [fn for fn in filenames]
 
 if len(checkpoint_files) > 0:
     model = sflax.MoDLNet(