From 58465e3bf5c00ee7d1909e2816d4f56238012026 Mon Sep 17 00:00:00 2001 From: suraj-self Date: Sat, 28 Feb 2026 13:10:50 +0530 Subject: [PATCH] fix: guard target-param-data-ratio against zero to avoid ZeroDivisionError --- scripts/base_train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/scripts/base_train.py b/scripts/base_train.py index 24091b6..9394950 100644 --- a/scripts/base_train.py +++ b/scripts/base_train.py @@ -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