From 9985e0bc08d65b22c345fdd12e0a9c018cb1a147 Mon Sep 17 00:00:00 2001
From: CaiLiYuan <74148589+diplomatist@users.noreply.github.com>
Date: Fri, 2 Dec 2022 16:31:27 +0800
Subject: [PATCH] Update yolov7_l_syncbn_fast_8x16b-300e_coco.py (#333)

change:model->bbox_head->head_module->dict(num_classes=num_classes)
---
 configs/yolov7/yolov7_l_syncbn_fast_8x16b-300e_coco.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/configs/yolov7/yolov7_l_syncbn_fast_8x16b-300e_coco.py b/configs/yolov7/yolov7_l_syncbn_fast_8x16b-300e_coco.py
index 5c4e5141..659f3a0f 100644
--- a/configs/yolov7/yolov7_l_syncbn_fast_8x16b-300e_coco.py
+++ b/configs/yolov7/yolov7_l_syncbn_fast_8x16b-300e_coco.py
@@ -67,7 +67,7 @@ model = dict(
         type='YOLOv7Head',
         head_module=dict(
             type='YOLOv7HeadModule',
-            num_classes=80,
+            num_classes=num_classes,
             in_channels=[256, 512, 1024],
             featmap_strides=strides,
             num_base_priors=3),