From 3515ddc2e9e09a70eb68718edfaea7896b77ea41 Mon Sep 17 00:00:00 2001
From: Ved Patwardhan <54766411+vedpatwardhan@users.noreply.github.com>
Date: Wed, 3 Jan 2024 21:56:18 +0530
Subject: [PATCH] fix: removed the additional permute dims from the forward
 pass of resnet as adaptive_avg_pool2d also has the NHWC data format

---
 ivy_models/resnet/resnet.py | 1 -
 1 file changed, 1 deletion(-)

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)