Skip to content

Commit efacc95

Browse files
authored
Update initialization to the normal xavier distribution (#221)
1 parent e21787e commit efacc95

File tree

1 file changed

+1
-1
lines changed
  • autoPyTorch/pipeline/components/setup/network_initializer

1 file changed

+1
-1
lines changed

autoPyTorch/pipeline/components/setup/network_initializer/XavierInit.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def initialization(m: torch.nn.Module) -> None:
2424
torch.nn.Conv2d,
2525
torch.nn.Conv3d,
2626
torch.nn.Linear)):
27-
torch.nn.init.xavier_uniform_(m.weight.data)
27+
torch.nn.init.xavier_normal(m.weight.data)
2828
if m.bias is not None and self.bias_strategy == 'Zero':
2929
torch.nn.init.constant_(m.bias.data, 0.0)
3030
return initialization

0 commit comments

Comments
 (0)