فهرست منبع

debug rtdetr_n on coco

yjh0410 2 سال پیش
والد
کامیت
fda13c908c
1فایلهای تغییر یافته به همراه4 افزوده شده و 2 حذف شده
  1. 4 2
      train.sh

+ 4 - 2
train.sh

@@ -3,15 +3,17 @@ python train.py \
         --cuda \
         -d coco \
         --root /mnt/share/ssd2/dataset/ \
-        -m yolox_m \
+        -m rtdetr_n \
         -bs 16 \
         -size 640 \
         --wp_epoch 3 \
-        --max_epoch 300 \
+        --max_epoch 150 \
         --eval_epoch 10 \
         --ema \
         --fp16 \
         --multi_scale \
+        --mosaic 0 \
+        --mixup 0
         # --resume weights/coco/yolox_s/yolox_s_best.pth \
         # --pretrained weights/coco/yolo_free_medium/yolo_free_medium_39.46.pth \
         # --eval_first