After static . quantization, I receive the following error-
RuntimeError: No function is registered for schema aten::thnn_conv2d_forward(Tensor self, Tensor weight, int[2] kernel_size, Tensor? bias, int[2] stride, int[2] padding) -> (Tensor output, Tensor finput, Tensor fgrad_input) on tensor type QuantizedCPUTensorId; available functions are CPUTensorId, VariableTensorId
raised by traced_script_module = torch.jit.trace(quantized, q_example, check_trace=False)
full code here - https://github.com/raghavgurbaxani/experiments/blob/master/try_static_quant.py 15