bugfix-missing activations in subnetwork

a3c/model.py

@@ -21,7 +21,9 @@ class FlatNet(nn.Module):
         
         self.feat_1 = nn.Sequential(
             nn.Conv2d(22, 128, kernel_size=3, padding=1),
+            nn.ReLU(True),
             nn.Conv2d(128, 128, kernel_size=3, padding=1),
+            nn.ReLU(True),
             nn.Conv2d(128, 128, kernel_size=3, padding=1),
             #nn.MaxPool2d(kernel_size=2, stride=2)
         )
@@ -31,6 +33,7 @@ class FlatNet(nn.Module):
         )
         self.feat_2 = nn.Sequential(
             nn.Conv2d(256, 256, kernel_size=3, padding=1),
+            nn.ReLU(True),
             nn.Conv2d(256, 256, kernel_size=3, padding=1),
            # nn.MaxPool2d(kernel_size=2, stride=2)
         )