

module_swap_list (Set[type]) -- list of float
        module types to attach the shadow

logger_cls (Callable) -- type of logger to be used in
    shadow module to process the outputs of quantized module and
    its float shadow module



torch.ao.ns._numeric_suite.compare_model_stub(float_model, q_model, module_swap_list, *data, logger_cls=)
Compare quantized module in a model with its floating point
   counterpart, feeding both of them the same input. Return a dict
   with key corresponding to module names and each entry being a
   dictionary with two keys 'float' and 'quantized', containing the
   output tensors of quantized and its matching float shadow module.
   This dict can be used to compare and compute the module level
   quantization error.
This function first call prepare_model_with_stubs() to swap the
   quantized module that we want to compare with the Shadow module,