Hello, I’d like to train a model in a two-stage fashion. In the first stage, I’d like to keep a certain subset of parameters fixed to initial values (say, all parameters in a certain Block). In the second stage, I’d like to train all parameters.
What is an easy way to exclude certain parameters (say, those of a Block, being part of a larger model) from training? Do I need to filter collect_params()? Example appreciated!