Update README and target epochs

hm-ysjiang · Jun 4, 2023 · 621b99a · 621b99a
1 parent 3fb73f7
commit 621b99a
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 10 deletions.
diff --git a/README.md b/README.md
@@ -11,6 +11,7 @@ Zachary Teed and Jia Deng<br/>
  - [ ] Train supervised to 200 (See if scratch is good enough)
  - [ ] Train self-supervised to 200 (Using scratch / transfer)
  - [ ] Run experiments of ctx size = { 96, 64 }
+ - [ ] SSIM error on supervised setting
  - [ ] Add self-collected dataset experiments
  - [ ] Try-out context attention module ([GMFlowNet](https://github.com/xiaofeng94/GMFlowNet))
 

diff --git a/train-selfsupervised.py b/train-selfsupervised.py
@@ -64,8 +64,9 @@ def sequence_loss(flow_preds: List[torch.Tensor], flow_gt: torch.Tensor,
 
     for i in range(n_predictions):
         i_weight = gamma**(n_predictions - i - 1)
-        l1_err, ssim_err = photometric_error(
-            image1, image2, flow_preds[i], valid[:, None])
+        l1_err, ssim_err = photometric_error(image1, image2,
+                                             flow_preds[i],
+                                             valid[:, None])
         i_loss = (1 - SSIM_WEIGHT) * l1_err + SSIM_WEIGHT * ssim_err
         flow_loss += i_weight * i_loss
 

diff --git a/train-selfsupervised.sh b/train-selfsupervised.sh
@@ -4,7 +4,7 @@ mkdir -p checkpoints
 cmd_scratch="python -u train-selfsupervised.py \
                     --name raft-sintel-selfsupervised-scratch \
                     --validation sintel \
-                    --num_epochs 100 \
+                    --num_epochs 200 \
                     --batch_size 6 \
                     --lr 0.0004 \
                     --wdecay 0.00001"
@@ -14,7 +14,7 @@ cmd_transfer="python -u train-selfsupervised.py \
                      --validation sintel \
                      --restore_ckpt checkpoints/raft-things.pth \
                      --freeze_bn \
-                     --num_epochs 100 \
+                     --num_epochs 200 \
                      --batch_size 6 \
                      --lr 0.000125 \
                      --wdecay 0.00001 \

diff --git a/train-supervised.sh b/train-supervised.sh
@@ -4,7 +4,7 @@ mkdir -p checkpoints
 cmd_scratch="python -u train-supervised.py \
                     --name raft-sintel-supervised-scratch \
                     --validation sintel \
-                    --num_epochs 100 \
+                    --num_epochs 200 \
                     --batch_size 6 \
                     --lr 0.0004 \
                     --wdecay 0.00001"
@@ -14,14 +14,14 @@ cmd_transfer="python -u train-supervised.py \
                      --validation sintel \
                      --restore_ckpt checkpoints/raft-things.pth \
                      --freeze_bn \
-                     --num_epochs 100 \
+                     --num_epochs 200 \
                      --batch_size 6 \
                      --lr 0.000125 \
                      --wdecay 0.00001 \
                      --gamma=0.85"
 
-# echo ${cmd_scratch}
-# eval ${cmd_scratch}
+echo ${cmd_scratch}
+eval ${cmd_scratch}
 
-echo ${cmd_transfer}
-eval ${cmd_transfer}
+# echo ${cmd_transfer}
+# eval ${cmd_transfer}