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

Compare commits

base: tacit:da0ac84e9e1b28ab1b254a26bf76da077f1c64a5
Branches Tags
tacit:master
...
compare: tacit:b113978aae5f29c0b2ac5221bad713d5b45c5020
Branches Tags
tacit:master

2 Commits
da0ac84e9e ... b113978aae

Author SHA1 Message Date
marked23
b113978aae
Merge b06bedac08 into 4a87a0d19f 2025-11-26 18:10:51 +01:00
Sofie Van Landeghem
b06bedac08
remove empty line 2025-11-26 18:09:23 +01:00
1 changed files with 0 additions and 1 deletions
Show Stats Expand all files Collapse all files

1
nanochat/checkpoint_manager.py
View File

@ -22,7 +22,6 @@ def log0(message):
def save_checkpoint(checkpoint_dir, step, model_data, optimizer_data, meta_data, rank=0): def save_checkpoint(checkpoint_dir, step, model_data, optimizer_data, meta_data, rank=0):
os.makedirs(checkpoint_dir, exist_ok=True) os.makedirs(checkpoint_dir, exist_ok=True)
if rank == 0: if rank == 0:
# Save the model state parameters # Save the model state parameters
model_path = os.path.join(checkpoint_dir, f"model_{step:06d}.pt") model_path = os.path.join(checkpoint_dir, f"model_{step:06d}.pt")