From b81d789992f42e6283ea7b0b980a0feb934b0bac Mon Sep 17 00:00:00 2001 From: Jason Kneen Date: Wed, 22 Oct 2025 09:29:46 +0100 Subject: [PATCH] Pass device batch size to base_loss script Added the --device_batch_size argument to the base_loss evaluation command in runmac_overnight.sh to ensure batch size is configurable during evaluation. --- dev/runmac_overnight.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dev/runmac_overnight.sh b/dev/runmac_overnight.sh index 31fad13..f270171 100755 --- a/dev/runmac_overnight.sh +++ b/dev/runmac_overnight.sh @@ -122,7 +122,7 @@ python -m scripts.base_train \ # Evaluate base model echo "" echo "Evaluating base model..." -python -m scripts.base_loss --split_tokens=$SPLIT_TOKENS +python -m scripts.base_loss --device_batch_size=$DEVICE_BATCH_SIZE --split_tokens=$SPLIT_TOKENS python -m scripts.base_eval # Midtraining