Merge pull request #148 from NKPmedia/return_path_not_filename

Return path not filename
delira-dev · Jun 17, 2019 · b6396a8 · b6396a8
2 parents 6feb661 + 3de40fc
commit b6396a8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/delira/training/base_trainer.py b/delira/training/base_trainer.py
@@ -775,10 +775,10 @@ def _search_for_prev_state(path, extensions=None):
                 int(x.rsplit("_", 1)[-1].rsplit(".", 1)[0])
                 for x in files])
 
-            latest_state_path = [x for x in files
+            latest_state_filename = [x for x in files
                                  if x.startswith("checkpoint_epoch_%d"
                                                  % latest_epoch)][0]
-
+            latest_state_path = os.path.join(path, latest_state_filename)
             return latest_state_path, latest_epoch
 
         return None, 1