From 715cb08b108b9f92519c5f2eeee6db27e3d9d930 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 8 Jun 2020 22:15:16 -0700
Subject: [PATCH] module updates

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

diff --git a/models/yolo.py b/models/yolo.py
index 9a5b8da..ae50c85 100644
--- a/models/yolo.py
+++ b/models/yolo.py
@@ -149,7 +149,7 @@ def parse_model(md, 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, ConvPlus, BottleneckCSP, BottleneckLight]:
+        if m in [nn.Conv2d, Conv, Bottleneck, SPP, DWConv, MixConv2d, Focus, ConvPlus, BottleneckCSP]:
             c1, c2 = ch[f], args[0]
 
             # Normal