yjh0410 2 éve
szülő
commit
4e667eb56f
1 módosított fájl, 3 hozzáadás és 3 törlés
  1. 3 3
      train_single_gpu.sh

+ 3 - 3
train_single_gpu.sh

@@ -3,14 +3,14 @@ python train.py \
         --cuda \
         -d coco \
         --root /data/datasets/ \
-        -m rtcdet_v2_l \
-        -bs 64 \
+        -m yolov7_tiny \
+        -bs 16 \
         -size 640 \
         --wp_epoch 3 \
         --max_epoch 300 \
         --eval_epoch 10 \
         --no_aug_epoch 15 \
-        --grad_accumulate 2 \
+        --grad_accumulate 1 \
         --ema \
         --fp16 \
         --multi_scale \