[FIXED] Weight initialization in PyTorch Lightning

Issue

Since PyTorch Lightning does most things under the hood, how can I know which weight initialization methods it use for training the model? And how can I modify the initialization method?

Solution

PyTorch Lightning doesn't interfere with weight initialization. It should be same as default PyTorch weights initialization. This is how Linear layers are initialized.

Answered By - Aniket Maurya