UPD: Those additional biases set to None
are becasue torch.backends.quantized.engine
mismath with torch.quantization.get_default_qat_qconfig
. When they match though (set to qnnpack
in my case) I get
RuntimeError: Tried to trace <__torch__.torch.classes.quantized.Conv2dPackedParamsBase object at 0x52982b0> but it is not part of the active trace. Modules that are called during a trace must be registered as submodules of the thing being traced.
1 Like