From b55ee49ca9b9417b65dc2c0654de2d2c55206269 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 20 Sep 2022 18:34:30 +0200 Subject: [PATCH 1/2] Update tf.py Signed-off-by: Glenn Jocher --- models/tf.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/models/tf.py b/models/tf.py index ae58ca738e2e..0f7d32c0811a 100644 --- a/models/tf.py +++ b/models/tf.py @@ -394,8 +394,7 @@ def parse_model(d, ch, model, imgsz): # 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, DWConv, DWConvTranspose2d, Bottleneck, SPP, SPPF, MixConv2d, Focus, CrossConv, + if m in [nn.Conv2d, Conv, DWConv, DWConvTranspose2d, Bottleneck, SPP, SPPF, MixConv2d, Focus, CrossConv, BottleneckCSP, C3, C3x]: c1, c2 = ch[f], args[0] c2 = make_divisible(c2 * gw, 8) if c2 != no else c2 From 83c0256580db42775540dc1b97b43c7695b1dfad Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 20 Sep 2022 16:35:03 +0000 Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- models/tf.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/models/tf.py b/models/tf.py index 0f7d32c0811a..ae58ca738e2e 100644 --- a/models/tf.py +++ b/models/tf.py @@ -394,7 +394,8 @@ def parse_model(d, ch, model, imgsz): # 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, DWConv, DWConvTranspose2d, Bottleneck, SPP, SPPF, MixConv2d, Focus, CrossConv, + if m in [ + nn.Conv2d, Conv, DWConv, DWConvTranspose2d, Bottleneck, SPP, SPPF, MixConv2d, Focus, CrossConv, BottleneckCSP, C3, C3x]: c1, c2 = ch[f], args[0] c2 = make_divisible(c2 * gw, 8) if c2 != no else c2