mirror of
https://github.com/karpathy/nanochat.git
synced 2026-04-02 13:45:21 +00:00
Fix formatting in wandb initialization line
This commit is contained in:
parent
a1fd43b34d
commit
1e989829b3
|
|
@ -61,7 +61,7 @@ dtype = torch.float32 if dtype == 'float32' else torch.bfloat16
|
|||
autocast_ctx = torch.amp.autocast(device_type="cuda", dtype=dtype)
|
||||
|
||||
# wandb logging init
|
||||
wandb_run = get_wandb("nanochat-rl" , run=run, master_process=master_process, user_config=user_config)
|
||||
wandb_run = get_wandb("nanochat-rl", run=run, master_process=master_process, user_config=user_config)
|
||||
|
||||
# Init model and tokenizer
|
||||
model, tokenizer, meta = load_model(source, device, phase="eval")
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user