Prefix default train output dir with current date
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
+3
-2
@@ -1,5 +1,5 @@
|
|||||||
from collections import Counter
|
from collections import Counter
|
||||||
from datetime import datetime, timezone
|
from datetime import date, datetime, timezone
|
||||||
from enum import Enum
|
from enum import Enum
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Optional
|
from typing import Optional
|
||||||
@@ -245,7 +245,8 @@ def train(
|
|||||||
)
|
)
|
||||||
|
|
||||||
out_dir = out or Path(
|
out_dir = out or Path(
|
||||||
f"checkpoints/{t['mode']}"
|
f"checkpoints/{date.today().strftime('%Y%m%d')}"
|
||||||
|
f"_{t['mode']}"
|
||||||
f"_h{m['hidden_dim']}"
|
f"_h{m['hidden_dim']}"
|
||||||
f"_b{m['n_blocks']}"
|
f"_b{m['n_blocks']}"
|
||||||
f"_e{m['emb_dim']}"
|
f"_e{m['emb_dim']}"
|
||||||
|
|||||||
Reference in New Issue
Block a user