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

Pass device batch size to base_loss script

Browse Source 
Added the --device_batch_size argument to the base_loss evaluation command in runmac_overnight.sh to ensure batch size is configurable during evaluation.
This commit is contained in:
Jason Kneen 2025-10-22 09:29:46 +01:00
parent 1225ddf00e
commit b81d789992
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dev/runmac_overnight.sh
View File

@ -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