fix ReLU

liouxy · Nov 25, 2017 · bf2d20d · bf2d20d
1 parent 48bb908
commit bf2d20d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model.py b/model.py
@@ -47,7 +47,7 @@ def __init__(self):
         )
 
         self.conv_output = nn.Conv2d(in_channels=64, out_channels=4, kernel_size=3, stride=1, padding=1)
-        self.relu = nn.Relu()
+        self.relu = nn.ReLU()
 
         for m in self.modules():
             if isinstance(m, nn.Conv2d):