QuantizedGraphModule

class torch.ao.quantization.fx.graph_module.QuantizedGraphModule(*args, **kwargs) [исходный код]

Этот класс создан для того, чтобы PackedParams (например, LinearPackedParams, Conv2dPackedParams) попадали в state_dict, позволяя сериализовать и десериализовать квантованный графовый модуль с помощью torch.save(m.state_dict()) и m.load_state_dict(state_dict).