when I use the hook to get output param, but the param’s type is Proxy.
def hook_get_output(module,  input, output):
    print(output)

for name, layer in module.named_modules():
     layer.register_forward_hook(hook_get_output)

notes: the module is quantization_module