There was an error while loading. Please reload this page.
1 parent b0a116e commit 5c2b513Copy full SHA for 5c2b513
super_resolution/model.py
@@ -24,7 +24,7 @@ def forward(self, x):
24
return x
25
26
def _initialize_weights(self):
27
- init.orthogonal(self.conv1.weight, init.gain('relu'))
28
- init.orthogonal(self.conv2.weight, init.gain('relu'))
29
- 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'))
30
init.orthogonal(self.conv4.weight)
0 commit comments