Skip to content

Commit a53d034

Browse files
Use /tmp for nanoGPT runtime defaults
1 parent fe5d0e3 commit a53d034

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

  • baseline/nanogpt_one_head/src/rg_nanogpt_one_head

baseline/nanogpt_one_head/src/rg_nanogpt_one_head/config.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -10,22 +10,22 @@
1010
import yaml
1111

1212
SUPPORTED_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

1616
def 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

0 commit comments

Comments
 (0)