Hey @saifkhanengr,
Please check this thread out. Also, check out the Version 12 of this kernel, which highlights the importance of the Input layer for weights initialization. Let us know if these help.
Cheers,
Elemento
1 Like