remove **kwargs from forward - _ConvBlock2D
@@ -35,7 +35,7 @@ class _ConvBlock2D(nn.Module):
self.batch_norm = _BatchNormZG(out_filters) if residual else nn.BatchNorm2d(out_filters) if\
batch_norm else lambda x: x
- def forward(self, input, **kwargs):
+ def forward(self, input):
res = input
input = self.conv(input, **kwargs)
input = self.dropout(input)
