Explorar el Código

train YOLOv5-N

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

+ 1 - 1
train.sh

@@ -12,7 +12,7 @@ python train.py \
         --ema \
         --fp16 \
         --multi_scale \
-        --resume weights/coco/yolov5_l/yolov5_l_epoch_211_46.56.pth \
+        # --resume weights/coco/yolov5_l/yolov5_l_epoch_211_46.56.pth \
         # --pretrained weights/coco/yolo_free_medium/yolo_free_medium_39.46.pth \
         # --eval_first