diff --git a/apex b/apex deleted file mode 160000 index 5b71d36..0000000 --- a/apex +++ /dev/null @@ -1 +0,0 @@ -Subproject commit 5b71d3695bf39efcdcda9dff5be2f70314b8f091 diff --git a/train_sf.sh b/train_sf.sh new file mode 100755 index 0000000..add95fe --- /dev/null +++ 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 +