Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix to rename --batches to --batch-size, former kept for BC

Browse source
This commit is contained in:
Yin Li 2020-09-12 18:21:51 -04:00
parent 7be3153206
commit 4799f8177c
5 changed files with 9 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

2
scripts/example-test.slurm
View file

@ -26,7 +26,7 @@ m2m.py test \
--test-tgt-patterns "test/D0-*.npy,test/D1-*.npy" \
--in-norms RnD.R0,RnD.R1 --tgt-norms RnD.D0,RnD.D1 \
--model model.Net --callback-at . \
--batches 1 \
--batch-size 1 \
--load-state checkpoint.pt

2
scripts/example-train.slurm
View file

@ -36,7 +36,7 @@ srun m2m.py train \
--val-tgt-patterns "val/D0-*.npy,val/D1-*.npy" \
--in-norms RnD.R0,RnD.R1 --tgt-norms RnD.D0,RnD.D1 \
--model model.Net --callback-at . \
--lr 1e-4 --batches 1 \
--lr 1e-4 --batch-size 1 \
--epochs 1024 --seed $RANDOM