Allow different weight initialization schemes for conv2d and linear. See: https://github.com/Kaixhin/nninit for some schemes.
Allow different weight initialization schemes for conv2d and linear.
See: https://github.com/Kaixhin/nninit for some schemes.