Remove MaxPool2d

lcswillems · Dec 30, 2019 · ab16cba · ab16cba
1 parent 6e1d0aa
commit ab16cba
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/model.py b/model.py
@@ -27,15 +27,14 @@ def __init__(self, obs_space, action_space, use_memory=False, use_text=False):
         self.image_conv = nn.Sequential(
             nn.Conv2d(3, 16, (2, 2)),
             nn.ReLU(),
-            nn.MaxPool2d((2, 2)),
             nn.Conv2d(16, 32, (2, 2)),
             nn.ReLU(),
             nn.Conv2d(32, 64, (2, 2)),
             nn.ReLU()
         )
         n = obs_space["image"][0]
         m = obs_space["image"][1]
-        self.image_embedding_size = ((n-1)//2-2)*((m-1)//2-2)*64
+        self.image_embedding_size = (n-3)*(m-3)*64
 
         # Define memory
         if self.use_memory: