Default weight initialization pytorch July 8, 2019 16 32 64 128 Pytorch default weight initialization conv2d Pytorch default weight initialization linear