File tree Expand file tree Collapse file tree
baseline/nanogpt_one_head/src/rg_nanogpt_one_head Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1010import yaml
1111
1212SUPPORTED_OPTIMIZERS = ("sgd_momentum" , "adamw" , "muon" )
13- DEFAULT_ROOT = Path . home () / " rg-nanogpt-one-head"
13+ DEFAULT_ROOT = Path ( "/tmp/ rg-nanogpt-one-head")
1414
1515
1616def roots () -> dict [str , Path ]:
17- root = Path (os .environ .get ("RG_NANOGPT_ONE_HEAD_ROOT" , DEFAULT_ROOT )). expanduser ()
17+ root = Path (os .environ .get ("RG_NANOGPT_ONE_HEAD_ROOT" , DEFAULT_ROOT ))
1818 return {
1919 "root" : root ,
2020 "data" : Path (
2121 os .environ .get ("RG_NANOGPT_ONE_HEAD_DATA_ROOT" , root / "data" )
22- ). expanduser () ,
22+ ),
2323 "results" : Path (
2424 os .environ .get ("RG_NANOGPT_ONE_HEAD_RESULTS_ROOT" , root / "results" )
25- ). expanduser () ,
25+ ),
2626 "plots" : Path (
2727 os .environ .get ("RG_NANOGPT_ONE_HEAD_PLOTS_ROOT" , root / "plots" )
28- ). expanduser () ,
28+ ),
2929 }
3030
3131
You can’t perform that action at this time.
0 commit comments