Merge pull request mkocabas#6 from protossw512/pull-request

fixed a bug in CoordConvTh with wrong argument in super call
HaoLiMRI · Jul 18, 2018 · 4279766 · 4279766
2 parents d8d6a2c + df32673
commit 4279766
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CoordConv.py b/CoordConv.py
@@ -57,7 +57,7 @@ def forward(self, input_tensor):
 class CoordConvTh(nn.Module):
     """CoordConv layer as in the paper."""
     def __init__(self, x_dim, y_dim, with_r, *args, **kwargs):
-        super(CoordConv, self).__init__()
+        super(CoordConvTh, self).__init__()
         self.addcoords = AddCoordsTh(x_dim=x_dim, y_dim=y_dim, with_r=with_r)
         self.conv = nn.Conv2d(*args, **kwargs)