lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix ConvTranspose weight init bug

This commit is contained in:
zhouwy19 2020-08-23 21:13:06 +08:00
parent 7e1678c7ea
commit e49a1cefc9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/jittor/nn.py
View File

@ -487,7 +487,7 @@ class ConvTranspose(Module):
self.dilation[1] * (self.kernel_size[1] - 1) - self.padding[1])
self.output_padding = output_padding if isinstance (output_padding, tuple) else (output_padding, output_padding)
self.weight = init.relu_invariant_gauss((in_channels, out_channels) + self.kernel_size, dtype="float", mode="fan_out")
self.weight = init.invariant_uniform((in_channels, out_channels) + self.kernel_size, dtype="float")
if bias:
self.bias = init.uniform([out_channels], dtype="float", low=-1, high=1)
else: