Selaa lähdekoodia

debug RTRDet-L on COCO

yjh0410 2 vuotta sitten
vanhempi
sitoutus
4fe7f17f2a
1 muutettua tiedostoa jossa 3 lisäystä ja 3 poistoa
  1. 3 3
      train_single_gpu.sh

+ 3 - 3
train_single_gpu.sh

@@ -2,11 +2,11 @@
 python train.py \
         --cuda \
         -d voc \
-        --root /mnt/share/ssd2/dataset/ \
-        -m rtcdet_n \
+        --root /data/datasets/ \
+        -m rtrdet_l \
         -bs 16 \
         -size 640 \
-        --wp_epoch 3 \
+        --wp_epoch 1 \
         --max_epoch 300 \
         --eval_epoch 10 \
         --no_aug_epoch 20 \