Added identity to support older versions of pytorch

Author: lukemelas

efficientnet_pytorch/utils.py

@@ -119,14 +119,22 @@ def __init__(self, in_channels, out_channels, kernel_size, image_size=None, **kw
         if pad_h > 0 or pad_w > 0:
             self.static_padding = nn.ZeroPad2d((pad_w // 2, pad_w - pad_w // 2, pad_h // 2, pad_h - pad_h // 2))
         else:
-            self.static_padding = nn.Identity()
+            self.static_padding = Identity()
 
     def forward(self, x):
         x = self.static_padding(x)
         x = F.conv2d(x, self.weight, self.bias, self.stride, self.padding, self.dilation, self.groups)
         return x
 
 
+class Identity(nn.Module):
+    def __init__(self,):
+        super(Identity, self).__init__()
+
+    def forward(self, input):
+        return input
+
+
 ########################################################################
 ############## HELPERS FUNCTIONS FOR LOADING MODEL PARAMS ##############
 ########################################################################