keep training YOLOvx-S

config/data_config/transform_config.py

@@ -303,8 +303,8 @@ yolovx_small_trans_config = {
     'aug_type': 'yolov5',
     # Basic Augment
     'degrees': 0.0,
-    'translate': 0.1,
-    'scale': 0.5,
+    'translate': 0.2,
+    'scale': 0.9,
     'shear': 0.0,
     'perspective': 0.0,
     'hsv_h': 0.015,
@@ -312,7 +312,7 @@ yolovx_small_trans_config = {
     'hsv_v': 0.4,
     # Mosaic & Mixup
     'mosaic_prob': 1.0,
-    'mixup_prob': 0.5,
+    'mixup_prob': 1.0,
     'mosaic_type': 'yolov5_mosaic',
     'mixup_type': 'yolox_mixup',
     'mixup_scale': [0.5, 1.5]   # "mixup_scale" is not used for YOLOv5MixUp

train_ddp.sh

@@ -16,3 +16,4 @@ python -m torch.distributed.run --nproc_per_node=8 train.py \
                                                     --fp16 \
                                                     --sybn \
                                                     --multi_scale \
+                                                    # --resume weights/coco/yolovx_s/yolox_s_best.pth \