fix passing value to ReLU

Nikronic · Jun 9, 2019 · e26cb0b · e26cb0b
1 parent 195542c
commit e26cb0b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/EdgeNet.py b/EdgeNet.py
@@ -16,7 +16,7 @@ def __init__(self, input_channel, output_channel):
 
         super(CBR, self).__init__()
         layers = [nn.Conv2d(input_channel, output_channel, kernel_size=5, stride=1, padding=2),
-                  nn.BatchNorm2d(num_features=output_channel), nn.ReLU(0.2, inplace=True)]
+                  nn.BatchNorm2d(num_features=output_channel), nn.ReLU(inplace=True)]
         self.layers = nn.Sequential(*layers)
 
     def forward(self, x):