fix: added dataformat for adaptive avgpool2d for alexnet

ivy-llc · Jan 3, 2024 · a24ac9a · a24ac9a
1 parent 3edcdff
commit a24ac9a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ivy_models/alexnet/alexnet.py b/ivy_models/alexnet/alexnet.py
@@ -41,7 +41,7 @@ def _build(self, *args, **kwargs):
             ivy.ReLU(),
             ivy.MaxPool2D(3, 2, 0, data_format="NCHW"),
         )
-        self.avgpool = ivy.AdaptiveAvgPool2d((6, 6))
+        self.avgpool = ivy.AdaptiveAvgPool2d((6, 6), data_format="NCHW")
         self.classifier = ivy.Sequential(
             ivy.Dropout(prob=self.spec.dropout),
             ivy.Linear(256 * 6 * 6, 4096),