Change "gain" -> "calculate_gain" (pytorch#192)

Fixes pytorch#189 (comment)
ZRStea · Aug 2, 2017 · 5c2b513 · 5c2b513
1 parent b0a116e
commit 5c2b513
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/super_resolution/model.py b/super_resolution/model.py
@@ -24,7 +24,7 @@ def forward(self, x):
         return x
 
     def _initialize_weights(self):
-        init.orthogonal(self.conv1.weight, init.gain('relu'))
-        init.orthogonal(self.conv2.weight, init.gain('relu'))
-        init.orthogonal(self.conv3.weight, init.gain('relu'))
+        init.orthogonal(self.conv1.weight, init.calculate_gain('relu'))
+        init.orthogonal(self.conv2.weight, init.calculate_gain('relu'))
+        init.orthogonal(self.conv3.weight, init.calculate_gain('relu'))
         init.orthogonal(self.conv4.weight)