This seems to be freezing weights layer-wase, i.e., all the params of a layer are frozen.
What I want is something more fine-tuned. For example, I’d like to identify top 5% of parameters (by their weights) in a given layer and only freeze those.