Append context and epoch info automatically

hm-ysjiang · Jun 7, 2023 · cec2a79 · cec2a79
1 parent e10af9b
commit cec2a79
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/train-selfsupervised.py b/train-selfsupervised.py
@@ -243,6 +243,7 @@ def train(args):
     args = parser.parse_args()
     if args.hidden != 128 or args.context != 128:
         args.reset_context = True
+    args.name = f'{args.name}-ep{args.num_epochs}-c{args.context}'
 
     torch.manual_seed(1234)
     np.random.seed(1234)

diff --git a/train-supervised.py b/train-supervised.py
@@ -237,6 +237,7 @@ def train(args):
     args = parser.parse_args()
     if args.hidden != 128 or args.context != 128:
         args.reset_context = True
+    args.name = f'{args.name}-ep{args.num_epochs}-c{args.context}'
 
     torch.manual_seed(1234)
     np.random.seed(1234)

diff --git a/train.sh b/train.sh
@@ -1,6 +1,6 @@
 # From scratch
 cmd_supervised_scratch="python -u train-supervised.py \
-                               --name supervised-scratch-c128 \
+                               --name supervised-scratch \
                                --validation sintel \
                                --num_epochs 100 \
                                --batch_size 3 \
@@ -9,7 +9,7 @@ cmd_supervised_scratch="python -u train-supervised.py \
                                --context 128"
 
 cmd_selfsupervised_scratch="python -u train-selfsupervised.py \
-                                   --name selfsupervised-scratch-c128 \
+                                   --name selfsupervised-scratch \
                                    --validation sintel \
                                    --num_epochs 100 \
                                    --batch_size 3 \
@@ -19,7 +19,7 @@ cmd_selfsupervised_scratch="python -u train-selfsupervised.py \
 
 # Transfer
 cmd_supervised_transfer="python -u train-supervised.py \
-                                --name supervised-transfer-c128 \
+                                --name supervised-transfer \
                                 --validation sintel \
                                 --restore_ckpt checkpoints/raft-things.pth \
                                 --num_epochs 100 \
@@ -32,7 +32,7 @@ cmd_supervised_transfer="python -u train-supervised.py \
                                 --context 128"
 
 cmd_selfsupervised_transfer="python -u train-selfsupervised.py \
-                                    --name selfsupervised-transfer-c128 \
+                                    --name selfsupervised-transfer \
                                     --validation sintel \
                                     --restore_ckpt checkpoints/raft-things.pth \
                                     --num_epochs 100 \