update

config/model_config/yolox_config.py

@@ -31,7 +31,7 @@ yolox_cfg = {
         'head_depthwise': False,
         # ---------------- Train config ----------------
         ## input
-        'multi_scale': [0.5, 1.25],   # 320 -> 960
+        'multi_scale': [0.7, 1.25],   # 448 -> 800
         'trans_type': 'yolox_nano',
         # ---------------- Assignment config ----------------
         ## matcher
@@ -75,7 +75,7 @@ yolox_cfg = {
         'head_depthwise': False,
         # ---------------- Train config ----------------
         ## input
-        'multi_scale': [0.5, 1.25],   # 320 -> 960
+        'multi_scale': [0.7, 1.25],   # 448 -> 800
         'trans_type': 'yolox_small',
         # ---------------- Assignment config ----------------
         ## matcher
@@ -119,7 +119,7 @@ yolox_cfg = {
         'head_depthwise': False,
         # ---------------- Train config ----------------
         ## input
-        'multi_scale': [0.5, 1.25],   # 320 -> 960
+        'multi_scale': [0.7, 1.25],   # 448 -> 800
         'trans_type': 'yolox_medium',
         # ---------------- Assignment config ----------------
         ## matcher
@@ -163,7 +163,7 @@ yolox_cfg = {
         'head_depthwise': False,
         # ---------------- Train config ----------------
         ## input
-        'multi_scale': [0.5, 1.25],   # 320 -> 800
+        'multi_scale': [0.7, 1.25],   # 448 -> 800
         'trans_type': 'yolox_large',
         # ---------------- Assignment config ----------------
         ## matcher
@@ -207,7 +207,7 @@ yolox_cfg = {
         'head_depthwise': False,
         # ---------------- Train config ----------------
         ## input
-        'multi_scale': [0.5, 1.25],   # 320 -> 800
+        'multi_scale': [0.7, 1.25],   # 448 -> 800
         'trans_type': 'yolox_huge',
         # ---------------- Assignment config ----------------
         ## matcher

train_ddp.sh

@@ -1,12 +1,12 @@
-# train YOLO with 8 GPUs
-# 使用 8 GPU来训练YOLO
-python -m torch.distributed.run --nproc_per_node=8 train.py \
+# train YOLO with 4 GPUs
+# 使用 4 GPU来训练YOLO
+python -m torch.distributed.run --nproc_per_node=4 train.py \
                                                     --cuda \
                                                     -dist \
                                                     -d coco \
                                                     --root /data/datasets/ \
-                                                    -m rtcdet_v2_n\
-                                                    -bs 128 \
+                                                    -m yolox_l\
+                                                    -bs 64 \
                                                     -size 640 \
                                                     --wp_epoch 3 \
                                                     --max_epoch 300 \