forked from XuelianCheng/LEAStereo
-
Notifications
You must be signed in to change notification settings - Fork 0
/
train_sf.sh
executable file
·15 lines (14 loc) · 905 Bytes
/
train_sf.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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