Hello. I am trying to post quantize a 3D ResNet using the new graph mode quantization. As far as I understand, both AdaptiveAvgPool3d and MaxPool3d have no quantized kernel and should be marked as “non traceable”. From here ((prototype) FX Graph Mode Quantization User Guide — PyTorch Tutorials 1.8.1+cu102 documentation), I thought the following should do the trick:
prep_config_dict = {
    "non_traceable_module_class": [nn.MaxPool3d, nn.AdaptiveAvgPool3d]
}
prepared_model = prepare_fx(
    model_to_quantize, qconfig_dict, prepare_custom_config_dict=prep_config_dict)

Yet, it does not. Neither did some experiments using “non_traceable_module_name”.
Appreciate any help I can get. Thank you in advance.