Fix an an untested change, remove a debug print

timm/models/efficientnet_blocks.py

@@ -220,7 +220,7 @@ class InvertedResidual(nn.Module):
         has_se = se_ratio is not None and se_ratio > 0.
         self.has_residual = (in_chs == out_chs and stride == 1) and not noskip
         self.drop_path_rate = drop_path_rate
-        print(act_layer)
+
         # Point-wise expansion
         self.conv_pw = create_conv2d(in_chs, mid_chs, exp_kernel_size, padding=pad_type, **conv_kwargs)
         self.bn1 = norm_layer(mid_chs, **norm_kwargs)

timm/models/layers/norm_act.py

@@ -60,7 +60,7 @@ class BatchNormAct2d(nn.BatchNorm2d):
         if torch.jit.is_scripting():
             x = self._forward_jit(x)
         else:
-            self._forward_python(x)
+            x = self._forward_python(x)
         if self.act is not None:
             x = self.act(x)
         return x