From 424b2b1b6a6719b325065f46b3a0d7410b21d8d3 Mon Sep 17 00:00:00 2001 From: XuelianCheng Date: Thu, 3 Dec 2020 16:03:40 +1100 Subject: [PATCH] add train_sf.sh --- apex | 1 - train_sf.sh | 15 +++++++++++++++ 2 files changed, 15 insertions(+), 1 deletion(-) delete mode 160000 apex create mode 100755 train_sf.sh 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 +