convert self.mask_kernel_numel to np.scalar
@@ -37,6 +37,7 @@ class PartialConv2d(nn.Conv2d):
self.kernel_size[1]))
self.mask_kernel_numel = np.prod(self.weight_mask_updater.shape[1:4])
+ self.mask_kernel_numel = np.asscalar(self.mask_kernel_numel)
def forward(self, input, mask=None, return_mask=True):
"""Forward function for partial conv2d.
