pytorch / TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
https://pytorch.org/TensorRT
BSD 3-Clause "New" or "Revised" License
2.6k stars 351 forks source link

🐛 [Bug] quantized_resnet_test failed #1641

Closed shuyuan-wang closed 1 year ago

shuyuan-wang commented 1 year ago

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:

  1. python examples/fx/quantized_resnet_test.py
截屏2023-02-01 16 14 57

Expected behavior

Environment

Build information about Torch-TensorRT can be found by turning on debug messages

Additional context

github-actions[bot] commented 1 year ago

This issue has not seen activity for 90 days, Remove stale label or comment or this will be closed in 10 days