forked from XuelianCheng/LEAStereo
-
Notifications
You must be signed in to change notification settings - Fork 0
/
train_kitti15.sh
20 lines (19 loc) · 1.12 KB
/
train_kitti15.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
CUDA_VISIBLE_DEVICES=0 python train.py \
--batch_size=4 \
--testBatchSize=1 \
--crop_height=288 \
--crop_width=576 \
--maxdisp=192 \
--threads=8 \
--dataset='kitti15' \
--save_path='./run/Kitti15/' \
--resume='./run/sceneflow/best/checkpoint/best.pth' \
--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=800 2>&1 |tee ./run/Kitti15/log.txt
#--resume='./run/Kitti15/best/best.pth' or './run/sceneflow/best/checkpoint/best.pth'