Skip to content

Commit 5c2b513

Browse files
Kaixhinsoumith
authored andcommitted
Change "gain" -> "calculate_gain" (#192)
Fixes #189 (comment)
1 parent b0a116e commit 5c2b513

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

super_resolution/model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def forward(self, x):
2424
return x
2525

2626
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'))
27+
init.orthogonal(self.conv1.weight, init.calculate_gain('relu'))
28+
init.orthogonal(self.conv2.weight, init.calculate_gain('relu'))
29+
init.orthogonal(self.conv3.weight, init.calculate_gain('relu'))
3030
init.orthogonal(self.conv4.weight)

0 commit comments

Comments
 (0)