Fix the checkpoint directory cleaning (#10)

eole-nlp · Jun 7, 2024 · 60fbbe4 · 60fbbe4
1 parent 9a299df
commit 60fbbe4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/eole/models/model_saver.py b/eole/models/model_saver.py
@@ -131,7 +131,9 @@ def __init__(
     def cleanup(self):
         if self.keep_checkpoint > 0:
             if len(self.checkpoint_queue) == self.checkpoint_queue.maxlen:
-                step_dir_to_delete = self.checkpoint_queue.popleft()
+                step_dir_to_delete = os.path.join(
+                    self.model_path, self.checkpoint_queue.popleft()
+                )
                 shutil.rmtree(step_dir_to_delete)
             self.checkpoint_queue.append(self.step_dir)