Explorar el Código

debug YOLOX-style Transform with Rotation

yjh0410 hace 2 años
padre
commit
a203d3c911
Se han modificado 1 ficheros con 3 adiciones y 3 borrados
  1. 3 3
      train.sh

+ 3 - 3
train.sh

@@ -7,9 +7,9 @@ python train.py \
         -bs 16 \
         -size 640 \
         --wp_epoch 1 \
-        --max_epoch 30 \
-        --eval_epoch 5 \
-        --no_aug_epoch 20 \
+        --max_epoch 5 \
+        --eval_epoch 1 \
+        --no_aug_epoch 2 \
         --ema \
         --fp16 \
         --multi_scale \