Use 'VALID' padding

Implementation detail from @mtthss
Kaixhin · Jun 17, 2019 · bdf8e39 · bdf8e39
1 parent 77d9e9c
commit bdf8e39
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/model.py b/model.py
@@ -51,13 +51,13 @@ def __init__(self, args, action_space):
     self.action_space = action_space
 
     if args.architecture == 'canonical':
-      self.convs = nn.Sequential(nn.Conv2d(args.history_length, 32, 8, stride=4, padding=1), nn.ReLU(),
-                                 nn.Conv2d(32, 64, 4, stride=2), nn.ReLU(),
-                                 nn.Conv2d(64, 64, 3, stride=1), nn.ReLU())
+      self.convs = nn.Sequential(nn.Conv2d(args.history_length, 32, 8, stride=4, padding=0), nn.ReLU(),
+                                 nn.Conv2d(32, 64, 4, stride=2, padding=0), nn.ReLU(),
+                                 nn.Conv2d(64, 64, 3, stride=1, padding=0), nn.ReLU())
       self.conv_output_size = 3136
     elif args.architecture == 'data-efficient':
-      self.convs = nn.Sequential(nn.Conv2d(args.history_length, 32, 5, stride=5, padding=1), nn.ReLU(),
-                                 nn.Conv2d(32, 64, 5, stride=5), nn.ReLU())
+      self.convs = nn.Sequential(nn.Conv2d(args.history_length, 32, 5, stride=5, padding=0), nn.ReLU(),
+                                 nn.Conv2d(32, 64, 5, stride=5, padding=0), nn.ReLU())
       self.conv_output_size = 576
     self.fc_h_v = NoisyLinear(self.conv_output_size, args.hidden_size, std_init=args.noisy_std)
     self.fc_h_a = NoisyLinear(self.conv_output_size, args.hidden_size, std_init=args.noisy_std)