Weight initilzation

Hi @Hamid,

I think you can extract the network’s parameters params = list(net.parameters()) and then apply the initialisation you may like.
If you need to apply the initialisation to a specific module, say conv1, you can extract the specific parameters with conv1Params = list(net.conv1.parameters()). You will have the kernels in conv1Params[0] and the bias terms in conv1Params[1].

6 Likes