update readme

README.md

@@ -237,3 +237,8 @@ python tools/infer.py --weights yolov6s6.pt --img 1280 --source img.jpg / imgdir
    </details>
 
 ### [FAQ（Continuously updated）](https://github.com/meituan/YOLOv6/wiki/FAQ%EF%BC%88Continuously-updated%EF%BC%89)
+
+If you have any questions, welcome to join our WeChat group to discuss and exchange.
+<p align="center">
+  <img src="assets/wechat_qrcode.png" align="middle" width = "1000" />
+</p>

README_cn.md

@@ -213,3 +213,8 @@ python tools/infer.py --weights yolov6s6.pt --img 1280 --source img.jpg / imgdir
 </details>
 
 ### [FAQ（持续更新）](https://github.com/meituan/YOLOv6/wiki/FAQ%EF%BC%88Continuously-updated%EF%BC%89)
+
+如果您有任何问题，欢迎加入我们的微信群一起讨论交流！
+<p align="center">
+  <img src="assets/wechat_qrcode.png" align="middle" width = "1000" />
+</p>

yolov6/core/engine.py

@@ -55,8 +55,8 @@ def __init__(self, args, cfg, device):
         self.distill_ns = True if self.args.distill and self.cfg.model.type in ['YOLOv6n','YOLOv6s'] else False
         model = self.get_model(args, cfg, self.num_classes, device)
         if self.args.distill:
-            if self.args.fuse_ab and self.distill_ns:
-                LOGGER.error('ERROR in: Distill n/s models should turn off the fuse_ab.\n')
+            if self.args.fuse_ab:
+                LOGGER.error('ERROR in: Distill models should turn off the fuse_ab.\n')
                 exit()
             self.teacher_model = self.get_teacher_model(args, cfg, self.num_classes, device)
         if self.args.quant: