Error I: Cannot find the extension library(_C.so)
Solution:
Error II: RuntimeError: Cannot load custom ops. Please rebuild the horizon_plugin_pytorch
Solution: check if the local CUDA environment is OK, such as path, version, etc.
RuntimeError: Only Tensors created explicitly by the user (graph leaves) support the deepcopy protocol at the moment
Solution: generally it is the inclusion of a non-leaf tensor in the model that causes this error, try the following:
TypeError: when calling function <built-in method conv2d of type object at >
Solution: the customized operator inherits the Module operator of a torch, which results in prepare_qat not being converted to a qat module. it is recommended to use the submodule method to call conv2d.
The QAT/Quantized accuracy is not as expected, there is a NAN, or the initial QAT loss is clearly anomalous with respect to float.
Solution: please refer to the section Accuracy Tuning Tool Guide.