pass | pass | ||||
n = max(round(n * gd), 1) if n > 1 else n # depth gain | 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] | c1, c2 = ch[f], args[0] | ||||
# Normal | # Normal |