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

Fix formatting in wandb initialization

Browse Source
This commit is contained in:
Sermet Pekin 2025-11-06 10:27:03 +03:00 committed by GitHub
parent 79e0eb8da9
commit 7a59092d8c
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
scripts/mid_train.py
View File

@ -64,7 +64,7 @@ synchronize = torch.cuda.synchronize if device_type == "cuda" else lambda: None
get_max_memory = torch.cuda.max_memory_allocated if device_type == "cuda" else lambda: 0
# wandb logging init
wandb_run = get_wandb("nanochat-mid" , run=run, master_process=master_process, user_config=user_config)
wandb_run = get_wandb("nanochat-mid", run=run, master_process=master_process, user_config=user_config)
# Load the model and tokenizer
model, tokenizer, meta = load_model("base", device, phase="train", model_tag=model_tag, step=step)