upgrade pytorch 0.4

Esse commit está contido em:
Robin Tibor Schirrmeister
2018-09-17 17:19:26 +02:00
commit d1bf8a5ca1
+1 -1
Ver Arquivo
@@ -17,7 +17,7 @@ def glorot_weight_zero_bias(model):
if not ('BatchNorm' in module.__class__.__name__):
init.xavier_uniform_(module.weight, gain=1)
else:
init.constant(module.weight, 1)
init.constant_(module.weight, 1)
if hasattr(module, 'bias'):
if module.bias is not None:
init.constant_(module.bias, 0)