`_initialize_weights` not working for the whole model

[Ngoc N. Tran]

By construction, the Sequential, SEModule, and Bottleneck modules are not custom-initialized.

[ShowLo]

By construction, the Sequential, SEModule, and Bottleneck modules are not custom-initialized.

Please refer to https://pytorch.org/docs/stable/generated/torch.nn.Module.html?highlight=modules#torch.nn.Module.modules, it will initialize the conv, linear and BN in Sequential, SEModule and Bottleneck.

[Ngoc N. Tran]

Well I'll be damned. Just tested and you're right, sorry for this confusing ticket.