Update lite model names

Meituan-AutoML · Aug 21, 2023 · a98bc2b · a98bc2b
1 parent ff84bdc
commit a98bc2b
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/configs/yolov6_lite/yolov6_lite_l_finetune.py b/configs/yolov6_lite/yolov6_lite_l_finetune.py
@@ -1,7 +1,7 @@
 # YOLOv6-lite-l model
 model = dict(
     type='YOLOv6-lite-l',
-    pretrained='weights/yolov6_lite_l.pt',
+    pretrained='weights/yolov6lite_l.pt',
     width_multiple=1.5,
     backbone=dict(
         type='Lite_EffiBackbone',

diff --git a/configs/yolov6_lite/yolov6_lite_m_finetune.py b/configs/yolov6_lite/yolov6_lite_m_finetune.py
@@ -1,7 +1,7 @@
 # YOLOv6-lite-m model
 model = dict(
     type='YOLOv6-lite-m',
-    pretrained='weights/yolov6_lite_m.pt',
+    pretrained='weights/yolov6lite_m.pt',
     width_multiple=1.1,
     backbone=dict(
         type='Lite_EffiBackbone',

diff --git a/configs/yolov6_lite/yolov6_lite_s_finetune.py b/configs/yolov6_lite/yolov6_lite_s_finetune.py
@@ -1,7 +1,7 @@
 # YOLOv6-lite-s model
 model = dict(
     type='YOLOv6-lite-s',
-    pretrained='weights/yolov6_lite_s.pt',
+    pretrained='weights/yolov6lite_s.pt',
     width_multiple=0.7,
     backbone=dict(
         type='Lite_EffiBackbone',