Update Middlebury fine-tuning.

princeton-vl · Mar 23, 2023 · 6068c1a · 6068c1a
1 parent fa8ed9d
commit 6068c1a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -138,7 +138,7 @@ chmod ug+x download_middlebury_2014.sh && ./download_middlebury_2014.sh
 ```
 and run
 ```Shell
-python train_stereo.py --train_datasets middlebury_2014 --num_steps 4000 --image_size 384 1000 --restore_ckpt models/raftstereo-sceneflow.pth --batch_size 2 --train_iters 22 --valid_iters 32 --spatial_scale -0.2 0.4 --saturation_range 0 1.4 --n_downsample 2  --mixed_precision
+python train_stereo.py --train_datasets middlebury_2014 --num_steps 4000 --image_size 384 1000 --lr 0.00002 --restore_ckpt models/raftstereo-sceneflow.pth --batch_size 2 --train_iters 22 --valid_iters 32 --spatial_scale -0.2 0.4 --saturation_range 0 1.4 --n_downsample 2  --mixed_precision
 ```
 
 ## (Optional) Faster Implementation

diff --git a/download_middlebury_2014.sh b/download_middlebury_2014.sh
@@ -1,4 +1,4 @@
-mkdir datasets/Middlebury/2014
+mkdir datasets/Middlebury/2014 -p
 cd datasets/Middlebury/2014
 wget https://vision.middlebury.edu/stereo/data/scenes2014/zip/Adirondack-imperfect.zip
 wget https://vision.middlebury.edu/stereo/data/scenes2014/zip/Adirondack-perfect.zip