diff --git a/ivy_models/resnet/resnet.py b/ivy_models/resnet/resnet.py index 469cb0e..dd65816 100644 --- a/ivy_models/resnet/resnet.py +++ b/ivy_models/resnet/resnet.py @@ -157,7 +157,6 @@ def _forward(self, x): x = self.layer2(x) x = self.layer3(x) x = self.layer4(x) - x = ivy.permute_dims(x, (0, 3, 1, 2)) x = self.avgpool(x) x = x.reshape((x.shape[0], -1)) x = self.fc(x)