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

fix: guard target-param-data-ratio against zero to avoid ZeroDivisionError

Browse Source
This commit is contained in:
suraj-self 2026-02-28 13:10:50 +05:30
parent c7ba252142
commit 58465e3bf5
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

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