tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2026-06-04 05:58:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge 15cc130034 into dc54a1a307

Browse Source
This commit is contained in:
Suraj-Self 2026-05-05 01:13:20 -04:00 committed by GitHub
commit 118926928d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
scripts/base_train.py
View File

@ -78,6 +78,7 @@ parser.add_argument("--save-every", type=int, default=-1, help="save checkpoints
# Output
parser.add_argument("--model-tag", type=str, default=None, help="override model tag for checkpoint directory name")
args = parser.parse_args()
assert args.target_param_data_ratio > 0 or args.target_param_data_ratio == -1, "target-param-data-ratio must be positive (or -1 to disable)"
user_config = vars(args).copy() # for logging
# -----------------------------------------------------------------------------
# Compute init and wandb logging