Merge pull request #28 from szupzp/master

Update network.py
nashory · Mar 8, 2018 · c128da6 · c128da6
2 parents 1ee7448 + 57cee22
commit c128da6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/network.py b/network.py
@@ -98,6 +98,7 @@ def intermediate_block(self, resl):
             halving = True
             for i in range(int(resl)-5):
                 ndim = ndim/2
+        ndim = int(ndim)
         layers = []
         layers.append(nn.Upsample(scale_factor=2, mode='nearest'))       # scale up by factor of 2.0
         if halving:
@@ -224,6 +225,7 @@ def intermediate_block(self, resl):
             halving = True
             for i in range(int(resl)-5):
                 ndim = ndim/2
+        ndim = int(ndim)
         layers = []
         if halving:
             layers = conv(layers, ndim, ndim, 3, 1, 1, self.flag_leaky, self.flag_bn, self.flag_wn, pixel=False)