Skip to content

Commit

Permalink
Unify batch size
Browse files Browse the repository at this point in the history
  • Loading branch information
@hm-ysjiang
hm-ysjiang committed Jun 7, 2023
1 parent 0c7a1e2 commit e10af9b
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions train.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ cmd_supervised_scratch="python -u train-supervised.py \
--name supervised-scratch-c128 \
--validation sintel \
--num_epochs 100 \
--batch_size 6 \
--batch_size 3 \
--lr 0.0004 \
--wdecay 0.00001 \
--context 128"
Expand All @@ -12,7 +12,7 @@ cmd_selfsupervised_scratch="python -u train-selfsupervised.py \
--name selfsupervised-scratch-c128 \
--validation sintel \
--num_epochs 100 \
--batch_size 6 \
--batch_size 3 \
--lr 0.0004 \
--wdecay 0.00001 \
--context 128"
Expand All @@ -23,7 +23,7 @@ cmd_supervised_transfer="python -u train-supervised.py \
--validation sintel \
--restore_ckpt checkpoints/raft-things.pth \
--num_epochs 100 \
--batch_size 6 \
--batch_size 3 \
--lr 0.0002 \
--wdecay 0.00001 \
--gamma=0.85 \
Expand All @@ -36,7 +36,7 @@ cmd_selfsupervised_transfer="python -u train-selfsupervised.py \
--validation sintel \
--restore_ckpt checkpoints/raft-things.pth \
--num_epochs 100 \
--batch_size 6 \
--batch_size 3 \
--lr 0.0002 \
--wdecay 0.00001 \
--gamma=0.85 \
Expand Down

0 comments on commit e10af9b

Please sign in to comment.