tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2026-03-27 23:25:14 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge 1bce71e03d into 1076f97059

Browse Source
This commit is contained in:
Suraj-Self 2026-03-05 16:35:57 +10:00 committed by GitHub
commit 81d3fccf87
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

@ -80,6 +80,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