Hello all ,
I want to specifically add the pretrained model parameters of some layers to my new network .
For Linear Layer i just did :
    model_enc.linear_3d.weight = model_trained.linear_3d.weight
    model_enc.linear_3d.bias = model_trained.linear_3d.bias

Will this suffice or are there any other parameters that I need to load or is there a easy way than this . My module is already trained and I just want to load params for few layers .
Thank you.