tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2026-02-09 12:09:49 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

replace f.name.split('.')[0] with f.stem

Browse Source 
Co-authored-by: Sofie Van Landeghem <svlandeg@users.noreply.github.com>
This commit is contained in:
Tsvika Shapira 2025-12-27 16:44:46 +02:00 committed by GitHub
parent b2dcc8dda7
commit b74444625c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nanochat/checkpoint_manager.py
View File

@ -119,7 +119,7 @@ def find_last_step(checkpoint_dir):
checkpoint_files = list(checkpoint_dir.glob("model_*.pt"))
if not checkpoint_files:
raise FileNotFoundError(f"No checkpoints found in {checkpoint_dir}")
last_step = int(max(f.name.split("_")[-1].split(".")[0] for f in checkpoint_files))
last_step = int(max(f.stem.split("_")[-1] for f in checkpoint_files))
return last_step
# -----------------------------------------------------------------------------