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]
.