Procházet zdrojové kódy

debug RTRDet-L on COCO

yjh0410 před 2 roky
rodič
revize
4fe7f17f2a
1 změnil soubory, kde provedl 3 přidání a 3 odebrání
  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 \