update

dvlab-research · Oct 31, 2023 · b7e7b7f · b7e7b7f
1 parent 42aeec4
commit b7e7b7f
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/config/ade20k/ade20k_release_psp18_psp101.yaml b/config/ade20k/ade20k_release_psp18_psp101.yaml
@@ -77,7 +77,6 @@ TEST:
   index_step: 0  # evaluation step index in list, 0 means to end
   test_gpu: [0]
   model_path: exp/release/ade20k_kd_psp18_psp101/model/train_epoch_100.pth  # evaluation model path
-  #model_path: initmodel/pspnet101_context_train_epoch_200.pth
   save_folder: exp/release/ade20k_kd_psp18_psp101/result/epoch_99/val/ss  # results save folder
   colors_path: data/ade20k/ade20k_colors.txt  # path of dataset colors
   names_path: data/ade20k/ade20k_names.txt  # path of dataset category names
diff --git a/tool/test.sh b/tool/test.sh
@@ -18,9 +18,9 @@ config=config/${dataset}/${dataset}_${exp_name}.yaml
 now=$(date +"%Y%m%d_%H%M%S")
 
 mkdir -p ${result_dir}
-cp tool/test_fc.sh tool/test_fc.py ${config} ${exp_dir}
+cp tool/test.sh tool/test.py ${config} ${exp_dir}
 
 export PYTHONPATH=./
-$PYTHON -u ${exp_dir}/test_fc.py \
+$PYTHON -u ${exp_dir}/test.py \
   --config=${config} \
   2>&1 | tee ${result_dir}/test-$now.log
diff --git a/tool/train.sh b/tool/train.sh
@@ -22,9 +22,9 @@ mkdir -p ${model_dir} ${result_dir}
 cp tool/train.py tool/train.sh tool/test.sh tool/test.py ${config} ${exp_dir}
 
 export PYTHONPATH=./
-#$PYTHON -u ${exp_dir}/train.py \
-#  --config=${config} \
-#  2>&1 | tee ${model_dir}/train-$now.log
+$PYTHON -u ${exp_dir}/train.py \
+  --config=${config} \
+  2>&1 | tee ${model_dir}/train-$now.log
 
 $PYTHON -u ${exp_dir}/test.py \
   --config=${config} \