From 7807ef5f38cb68194787133ed875310d695d2970 Mon Sep 17 00:00:00 2001
From: Paul Best <paul.best@lis-lab.fr>
Date: Thu, 5 Jan 2023 16:34:12 +0100
Subject: [PATCH] fix maxPool for orca and globi
---
models.py | 2 --
1 file changed, 2 deletions(-)
diff --git a/models.py b/models.py
index 488f054..b046caf 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))
)
)
}
--
GitLab