Merge 958543fcd5 into 4a87a0d19f

Co-authored-by: Sofie Van Landeghem <svlandeg@users.noreply.github.com>

nanochat/checkpoint_manager.py

@@ -117,7 +117,7 @@ def find_last_step(checkpoint_dir):
     checkpoint_files = [f for f in os.listdir(checkpoint_dir) if re.search(r'model_(\d+)\.pt$', f)]
     if not checkpoint_files:
         raise FileNotFoundError(f"No checkpoints found in {checkpoint_dir}")
-    last_step = int(max(re.search(r'model_(\d+)\.pt$', f).group(1) for f in checkpoint_files))
+    last_step = int(max(f.split("_")[-1].split(".")[0] for f in checkpoint_files))
     return last_step
 
 # -----------------------------------------------------------------------------