Switching off training of certain weights

[kaustubhjsawant]

Hello,
I have been using Nequip for some time now and was curious if there is any obvious way to switch off training for specific weights in the network.
I have a large dataset A and wanted to train on a new smaller dataset B which is very similar to A. Does it make sense to train a large network on A and train only certain weights of the pre-trained network on B (in some sense finetuning the network)?

Thank you,
Kaustubh Sawant