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

Fix formatting of wandb_run initialization

Browse Source
This commit is contained in:
Sermet Pekin 2025-11-06 10:24:04 +03:00 committed by GitHub
parent 545bb8e772
commit a1fd43b34d
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/base_train.py
View File

@ -74,7 +74,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",run=run, master_process=master_process, user_config=user_config)
wandb_run = get_wandb("nanochat", run=run, master_process=master_process, user_config=user_config)
# Tokenizer will be useful for evaluation, also we need the vocab size
tokenizer = get_tokenizer()