Load a single layer weights from a pretrained model

autograd
1

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.