diff --git a/models.py b/models.py
index 488f054e5682962f3cba8c19a7df2733d08bf298..b046cafaa3bca3f1592ebe06fc39d17894e36fbf 100644
--- a/models.py
+++ b/models.py
@@ -181,7 +181,6 @@ get = {
                 nn.Dropout2d(p=.5),
                 nn.LeakyReLU(0.01),
                 nn.Conv2d(64, 1, 1, bias=False),
-                nn.AdaptiveMaxPool2d(output_size=(1, 1))
             )
         )
     },
@@ -223,7 +222,6 @@ get = {
                 nn.Dropout(p=.5),
                 nn.Conv2d(64, 1, 1, bias=False),
                 nn.MaxPool2d((6, 1)),
-                nn.AdaptiveMaxPool2d(output_size=(1, 1))
             )
         )
     }