fix minor error

qjadud1994 · Apr 14, 2021 · e7a82f5 · e7a82f5
1 parent 3f840be
commit e7a82f5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/run.sh b/run.sh
@@ -4,7 +4,7 @@ CUDA_VISIBLE_DEVICES=0,1 python scripts/train_cls.py \
     --lr=0.001 \
     --epoch=15 \
     --decay_points='5,10' \
-    --thresh=0 \
+    --delta=0 \
     --logdir=logs/DRS_learnable \
     --save_folder=checkpoints/DRS_learnable \
     --show_interval=50
@@ -40,4 +40,4 @@ CUDA_VISIBLE_DEVICES=0,1 python scripts/train_refine.py \
 CUDA_VISIBLE_DEVICES=0 python scripts/pseudo_seg_label_gen.py \
     --img_dir=/data/DB/VOC2012/ \
     --checkpoint=checkpoints/Refine_DRS_learnable/best.pth
-
+