yjh0410 hace 2 años
padre
commit
4e667eb56f
Se han modificado 1 ficheros con 3 adiciones y 3 borrados
  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 \