From 35dc9355044b8d94d7e6a9d43677edcc569ad7d0 Mon Sep 17 00:00:00 2001
From: tomguluson92 <314913739@qq.com>
Date: Sat, 12 Sep 2020 07:35:48 +0800
Subject: [PATCH] remove `nn.Conv2d` in default settings (#949)

---
 models/yolo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/yolo.py b/models/yolo.py
index f1c3c3f..1a9a2c6 100644
--- a/models/yolo.py
+++ b/models/yolo.py
@@ -188,7 +188,7 @@ def parse_model(d, ch):  # model_dict, input_channels(3)
                 pass
 
         n = max(round(n * gd), 1) if n > 1 else n  # depth gain
-        if m in [nn.Conv2d, Conv, Bottleneck, SPP, DWConv, MixConv2d, Focus, CrossConv, BottleneckCSP, C3]:
+        if m in [Conv, Bottleneck, SPP, DWConv, MixConv2d, Focus, CrossConv, BottleneckCSP, C3]:
             c1, c2 = ch[f], args[0]
 
             # Normal