-
Notifications
You must be signed in to change notification settings - Fork 51
/
train_kitti12.sh
20 lines (19 loc) · 1.08 KB
/
train_kitti12.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='kitti12' \
--save_path='./run/Kitti12/' \
--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/Kitti12/log.txt
#--resume='./run/Kitti12/best/best_1.16.pth'