Thanks for the additional info.  Looks like what you are describing was not supported, and added recently with https://github.com/pytorch/pytorch/pull/48072 1.  Before that PR, the type(mod) == FakeQuantize check would not pass for custom fake quant classes.  It should pass now.  Could you try updating your PyTorch installation and see if it works with 1.7.1 or master?