From 41bbb3588303fbebb64d939005b2c838968c5a30 Mon Sep 17 00:00:00 2001 From: hm-ysjiang Date: Wed, 7 Jun 2023 13:53:46 +0800 Subject: [PATCH] Training Commands --- train.sh | 48 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 48 insertions(+) create mode 100644 train.sh diff --git a/train.sh b/train.sh new file mode 100644 index 0000000..d5df6bd --- /dev/null +++ b/train.sh @@ -0,0 +1,48 @@ +# From scratch +cmd_supervised_scratch="python -u train-supervised.py \ + --name supervised-scratch-c128 \ + --validation sintel \ + --num_epochs 100 \ + --batch_size 6 \ + --lr 0.0004 \ + --wdecay 0.00001 \ + --context 128" + +cmd_selfsupervised_scratch="python -u train-selfsupervised.py \ + --name selfsupervised-scratch-c128 \ + --validation sintel \ + --num_epochs 100 \ + --batch_size 6 \ + --lr 0.0004 \ + --wdecay 0.00001 \ + --context 128" + +# Transfer +cmd_supervised_transfer="python -u train-supervised.py \ + --name supervised-transfer-c128 \ + --validation sintel \ + --restore_ckpt checkpoints/raft-things.pth \ + --num_epochs 100 \ + --batch_size 6 \ + --lr 0.0002 \ + --wdecay 0.00001 \ + --gamma=0.85 \ + --reset_context \ + --context 128" + +cmd_selfsupervised_transfer="python -u train-selfsupervised.py \ + --name selfsupervised-transfer-c128 \ + --validation sintel \ + --restore_ckpt checkpoints/raft-things.pth \ + --num_epochs 100 \ + --batch_size 6 \ + --lr 0.0002 \ + --wdecay 0.00001 \ + --gamma=0.85 \ + --reset_context \ + --context 128" + +cmd=$cmd_supervised_transfer # Change this line + +echo ${cmd} +eval ${cmd} \ No newline at end of file