Fix a typo in the definition of the network

stan-stripe · Feb 13, 2017 · 7cb36fb · 7cb36fb
1 parent e572b0a
commit 7cb36fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model.py b/model.py
@@ -36,7 +36,7 @@ class ActorCritic(torch.nn.Module):
 
     def __init__(self, num_inputs, action_space):
         super(ActorCritic, self).__init__()
-        self.conv1 = nn.Conv2d(num_inputs, 32, 3, stride=2)
+        self.conv1 = nn.Conv2d(num_inputs, 32, 3, stride=2, padding=1)
         self.conv2 = nn.Conv2d(32, 32, 3, stride=2, padding=1)
         self.conv3 = nn.Conv2d(32, 32, 3, stride=2, padding=1)
         self.conv4 = nn.Conv2d(32, 32, 3, stride=2, padding=1)