tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2025-12-14 16:22:13 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge 01f5f10122 into 4a87a0d19f

Browse Source
This commit is contained in:
Pyry Takala 2025-11-19 18:34:27 -08:00 committed by GitHub
commit d7f76e6fdc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
nanochat/checkpoint_manager.py
View File

@ -118,7 +118,16 @@ def find_last_step(checkpoint_dir):
checkpoint_files = glob.glob(os.path.join(checkpoint_dir, "model_*.pt"))
if not checkpoint_files:
raise FileNotFoundError(f"No checkpoints found in {checkpoint_dir}")
last_step = int(max(os.path.basename(f).split("_")[-1].split(".")[0] for f in checkpoint_files))
# Use regex to match only valid checkpoint files (model_<digits>.pt) and ignore malformed files
# This prevents crashes when files like model_000200_backup.pt exist in the directory
steps = []
for f in checkpoint_files:
match = re.match(r"model_(\d+)\.pt$", os.path.basename(f))
if match:
steps.append(int(match.group(1)))
if not steps:
raise ValueError(f"No valid checkpoint files found in {checkpoint_dir}")
last_step = max(steps)
return last_step
# -----------------------------------------------------------------------------