We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e21787e commit efacc95Copy full SHA for efacc95
autoPyTorch/pipeline/components/setup/network_initializer/XavierInit.py
@@ -24,7 +24,7 @@ def initialization(m: torch.nn.Module) -> None:
24
torch.nn.Conv2d,
25
torch.nn.Conv3d,
26
torch.nn.Linear)):
27
- torch.nn.init.xavier_uniform_(m.weight.data)
+ torch.nn.init.xavier_normal(m.weight.data)
28
if m.bias is not None and self.bias_strategy == 'Zero':
29
torch.nn.init.constant_(m.bias.data, 0.0)
30
return initialization
0 commit comments