debug train

config/yolov1_config.py

@@ -18,7 +18,7 @@ yolov1_cfg = {
     # model EMA
     'ema_decay': 0.9999,       # SGD: 0.9999;   AdamW: 0.9998
     # lr schedule
-    'scheduler': 'step',
+    'scheduler': 'linear',
     'lr0': 0.001,              # SGD: 0.01;     AdamW: 0.004
     'lrf': 0.01,               # SGD: 0.01;     AdamW: 0.05
     # warmup strategy

train.sh

@@ -0,0 +1,33 @@
+# Train FreeYOLO
+python train.py \
+        --cuda \
+        -d voc \
+        --root /mnt/share/ssd2/dataset/ \
+        -v yolov1 \
+        -bs 16 \
+        -size 640 \
+        --wp_epoch 1 \
+        --max_epoch 150 \
+        --step_epoch 90 120 \
+        --eval_epoch 10 \
+        --ema \
+        --fp16 \
+        --multi_scale \
+        # --resume weights/coco/yolo_free_vx_pico/yolo_free_vx_pico_epoch_41_20.46.pth \
+        # --pretrained weights/coco/yolo_free_medium/yolo_free_medium_39.46.pth \
+        # --eval_first
+
+# # Debug FreeYOLO on VOC
+# python train.py \
+#         --cuda \
+#         -d voc \
+#         --root /mnt/share/ssd2/dataset/ \
+#         -v yolo_free_v2_tiny \
+#         -bs 16 \
+#         --max_epoch 25 \
+#         --wp_epoch 1 \
+#         --eval_epoch 5 \
+#         --ema \
+#         --fp16 \
+#         # --resume weights/coco/yolo_free_medium/yolo_free_medium_epoch_31_39.46.pth \
+#         # --pretrained weights/coco/yolo_free_medium/yolo_free_medium_39.46.pth \