debug YOLOv5-Plus on VOC

models/detectors/yolov5_plus/loss.py

@@ -18,7 +18,6 @@ class Criterion(object):
         self.device = device
         self.num_classes = num_classes
         self.warmup_epoch = warmup_epoch
-        self.warmup_stage = True
         # ------------------ Loss Parameters ------------------
         ## loss function
         self.cls_lossf = ClassificationLoss(cfg, reduction='none')

train.sh

@@ -1,7 +1,7 @@
 # Train YOLO
 python train.py \
         --cuda \
-        -d coco \
+        -d voc \
         --root /mnt/share/ssd2/dataset/ \
         -m yolov5_plus_n \
         -bs 16 \