mirror of
https://github.com/karpathy/nanochat.git
synced 2025-12-06 04:12:13 +00:00
Compare commits
2 Commits
da0ac84e9e
...
b113978aae
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
b113978aae | ||
|
|
b06bedac08 |
|
|
@ -22,7 +22,6 @@ def log0(message):
|
|||
|
||||
def save_checkpoint(checkpoint_dir, step, model_data, optimizer_data, meta_data, rank=0):
|
||||
os.makedirs(checkpoint_dir, exist_ok=True)
|
||||
|
||||
if rank == 0:
|
||||
# Save the model state parameters
|
||||
model_path = os.path.join(checkpoint_dir, f"model_{step:06d}.pt")
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user