I have another question:
Because of the quantization operation ,I need to modify some layers’ grad.I want to use register_hook function.
To modify grad ,I will use the mask_weight and mask_bias as mentioned earilier .
How can register_hook register a function which need parameters?