Fix bug of training V6-RepOpt: no attribute 'atss_warmup_epoch'

naive0409 · Jan 12, 2023 · 7fa9ba9 · 7fa9ba9
1 parent 7dba38e
commit 7fa9ba9
Show file tree

Hide file tree

Showing 10 changed files with 10 additions and 1 deletion.
diff --git a/configs/repopt/yolov6_tiny_hs.py b/configs/repopt/yolov6_tiny_hs.py
@@ -22,6 +22,7 @@
         anchors=1,
         out_indices=[17, 20, 23],
         strides=[8, 16, 32],
+        atss_warmup_epoch=0,
         iou_type='siou',
         use_dfl=False,
         reg_max=0 #if use_dfl is False, please set reg_max to 0

diff --git a/configs/repopt/yolov6_tiny_opt.py b/configs/repopt/yolov6_tiny_opt.py
@@ -23,6 +23,7 @@
         anchors=1,
         out_indices=[17, 20, 23],
         strides=[8, 16, 32],
+        atss_warmup_epoch=0,
         iou_type='siou',
         use_dfl=False,
         reg_max=0 #if use_dfl is False, please set reg_max to 0

diff --git a/configs/repopt/yolov6_tiny_opt_qat.py b/configs/repopt/yolov6_tiny_opt_qat.py
@@ -23,6 +23,7 @@
         anchors=1,
         out_indices=[17, 20, 23],
         strides=[8, 16, 32],
+        atss_warmup_epoch=0,
         iou_type='siou',
         use_dfl=False,
         reg_max=0, #if use_dfl is False, please set reg_max to 0

diff --git a/configs/repopt/yolov6n_hs.py b/configs/repopt/yolov6n_hs.py
@@ -22,6 +22,7 @@
         anchors=1,
         out_indices=[17, 20, 23],
         strides=[8, 16, 32],
+        atss_warmup_epoch=0,
         iou_type='siou',
         use_dfl=False,
         reg_max=0 #if use_dfl is False, please set reg_max to 0

diff --git a/configs/repopt/yolov6n_opt.py b/configs/repopt/yolov6n_opt.py
@@ -23,6 +23,7 @@
         anchors=1,
         out_indices=[17, 20, 23],
         strides=[8, 16, 32],
+        atss_warmup_epoch=0,
         iou_type='siou',
         use_dfl=False,
         reg_max=0 #if use_dfl is False, please set reg_max to 0

diff --git a/configs/repopt/yolov6n_opt_qat.py b/configs/repopt/yolov6n_opt_qat.py
@@ -23,6 +23,7 @@
         anchors=1,
         out_indices=[17, 20, 23],
         strides=[8, 16, 32],
+        atss_warmup_epoch=0,
         iou_type='siou',
         use_dfl=False,
         reg_max=0, #if use_dfl is False, please set reg_max to 0

diff --git a/configs/repopt/yolov6s_hs.py b/configs/repopt/yolov6s_hs.py
@@ -22,6 +22,7 @@
         anchors=1,
         out_indices=[17, 20, 23],
         strides=[8, 16, 32],
+        atss_warmup_epoch=0,
         iou_type='giou',
         use_dfl=False,
         reg_max=0

diff --git a/configs/repopt/yolov6s_opt.py b/configs/repopt/yolov6s_opt.py
@@ -23,6 +23,7 @@
         anchors=1,
         out_indices=[17, 20, 23],
         strides=[8, 16, 32],
+        atss_warmup_epoch=0,
         iou_type='giou',
         use_dfl=False,
         reg_max=0

diff --git a/configs/repopt/yolov6s_opt_qat.py b/configs/repopt/yolov6s_opt_qat.py
@@ -23,6 +23,7 @@
         anchors=1,
         out_indices=[17, 20, 23],
         strides=[8, 16, 32],
+        atss_warmup_epoch=0,
         iou_type = 'giou',
         use_dfl = False,
         reg_max = 0,  # if use_dfl is False, please set reg_max to 0

diff --git a/configs/yolov6n.py b/configs/yolov6n.py
@@ -1,4 +1,4 @@
-# YOLOv6s model
+# YOLOv6n model
 model = dict(
     type='YOLOv6n',
     pretrained=None,