"git@developer.sourcefind.cn:OpenDAS/torch-harmonics.git" did not exist on "214fa40aba94468ab4ddd50f4b3544b1780b87b3"
Commit 74d04d2c authored by Yuanzheng Ci's avatar Yuanzheng Ci Committed by Soumith Chintala
Browse files

Change division style to prevent error in python3 (#127)

Using python2 style '/' will convert int type to float in python3 which will cause the following error when creating FloatTensor:

TypeError: torch.FloatTensor constructor received an invalid combination of arguments - got (float, int, int, int), but expected one of:
 * no arguments
 * (int ...)
      didn't match because some of the arguments have invalid types: (float, int, int, int)
parent 8f6b9df8
...@@ -139,9 +139,9 @@ class DenseNet(nn.Module): ...@@ -139,9 +139,9 @@ class DenseNet(nn.Module):
self.features.add_module('denseblock%d' % (i + 1), block) self.features.add_module('denseblock%d' % (i + 1), block)
num_features = num_features + num_layers * growth_rate num_features = num_features + num_layers * growth_rate
if i != len(block_config) - 1: if i != len(block_config) - 1:
trans = _Transition(num_input_features=num_features, num_output_features=num_features / 2) trans = _Transition(num_input_features=num_features, num_output_features=num_features // 2)
self.features.add_module('transition%d' % (i + 1), trans) self.features.add_module('transition%d' % (i + 1), trans)
num_features = num_features / 2 num_features = num_features // 2
# Final batch norm # Final batch norm
self.features.add_module('norm5', nn.BatchNorm2d(num_features)) self.features.add_module('norm5', nn.BatchNorm2d(num_features))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment