What is the easiest way to quantize only part of the model?

[Aure20]

I am trying to modify the code such that given for example a new strategy I can quantize the model only for the layers that are modified and not from scratch entirely (and probably also the one that come after since they are dependent on it). What would be the easiest way to achieve that?