changed batch size for example script

nasa-nccs-hpda · Aug 27, 2024 · 7e72fd7 · 7e72fd7
1 parent f433e11
commit 7e72fd7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/runs/runners/frontier_svtoa_pretraining_runner.sh b/runs/runners/frontier_svtoa_pretraining_runner.sh
@@ -89,7 +89,7 @@ echo $MASTER_PORT
 nnodes=$SLURM_JOB_NUM_NODES
 datapaths=/lustre/orion/geo160/proj-shared/data/satvision-toa/50m
 validationpath=/lustre/orion/geo160/proj-shared/data/satvision-toa/validation/sv_toa_128_chip_validation_04_24.npy
-batchsize=1536
+batchsize=256
 nprocpernode=8
 
 launcher="python -u -m torch.distributed.run --nnodes=${nnodes} --master_addr ${MASTER_ADDR} --master_port ${MASTER_PORT} --nproc_per_node=${nprocpernode}"