When I try to run example/fx/quantized_resnet_test.py, it gives AssertionError which assert engine failed. The error occurs because it failed to find scale for int8 mode.
To Reproduce
Steps to reproduce the behavior:
python examples/fx/quantized_resnet_test.py
Expected behavior
Environment
Build information about Torch-TensorRT can be found by turning on debug messages
Torch-TensorRT Version (e.g. 1.0.0):1.4.0.dev0+0c6b37d3
PyTorch Version (e.g. 1.0):1.13.1
CPU Architecture:x86_64
OS (e.g., Linux):ubuntu
How you installed PyTorch (conda, pip, libtorch, source):pip
Build command you used (if compiling from source):python setup.py install --fx-only
Are you using local sources or building from archives:
Bug Description
When I try to run example/fx/quantized_resnet_test.py, it gives AssertionError which assert engine failed. The error occurs because it failed to find scale for int8 mode.
To Reproduce
Steps to reproduce the behavior:
python examples/fx/quantized_resnet_test.py
Expected behavior
Environment
conda
,pip
,libtorch
, source):pipAdditional context