add train_sf.sh

XuelianCheng · Dec 3, 2020 · 424b2b1 · 424b2b1
1 parent 011254c
commit 424b2b1
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/apex b/apex
diff --git a/train_sf.sh b/train_sf.sh
@@ -0,0 +1,15 @@
+CUDA_VISIBLE_DEVICES=0 python3 train.py --batch_size=4 \
+                --crop_height=384 \
+                --crop_width=576 \
+                --maxdisp=192 \
+                --threads=8 \
+                --save_path='./run/sceneflow/retrain/' \
+                --fea_num_layer 6 --mat_num_layers 12 \
+                --fea_filter_multiplier 8 --fea_block_multiplier 4 --fea_step 3  \
+                --mat_filter_multiplier 8 --mat_block_multiplier 4 --mat_step 3  \
+                --net_arch_fea='run/sceneflow/best/architecture/feature_network_path.npy' \
+                --cell_arch_fea='run/sceneflow/best/architecture/feature_genotype.npy' \
+                --net_arch_mat='run/sceneflow/best/architecture/matching_network_path.npy' \
+                --cell_arch_mat='run/sceneflow/best/architecture/matching_genotype.npy' \
+                --nEpochs=20 2>&1 |tee ./run/sceneflow/retrain/log.txt
+