Description
Bug Description
FAILED models/test_reexport.py::test_base_full_compile - AssertionError: False is not true : test_base_model_full_compile TRT outputs don't match with the original model. Cosine sim score: nan Threshold: 0.99
FAILED models/test_reexport.py::test_base_full_compile_multiple_outputs - AssertionError: False is not true : test_base_full_compile_multiple_outputs TRT outputs don't match with the original model. Cosine sim score: nan Threshold: 0.99
FAILED models/test_reexport.py::test_hybrid_relu_fallback - AssertionError: False is not true : test_hybrid_relu_fallback TRT outputs don't match with the original model. Cosine sim score: nan Threshold: 0.99
FAILED models/test_reexport.py::test_resnet18 - AssertionError: False is not true : test_resnet18 TRT outputs don't match with the original model. Cosine sim score: 0.048845380544662476 Threshold: 0.99
FAILED models/test_reexport.py::test_resnet18_dynamic - KeyError: u0
FAILED models/test_reexport.py::test_resnet18_dynamic_fallback - KeyError: u0
FAILED models/test_reexport.py::test_random_dynamic_fallback - KeyError: u0
To Reproduce
Steps to reproduce the behavior:
Expected behavior
Environment
Build information about Torch-TensorRT can be found by turning on debug messages
- Torch-TensorRT Version (e.g. 1.0.0):
- PyTorch Version (e.g. 1.0):
- CPU Architecture:
- OS (e.g., Linux):
- How you installed PyTorch (
conda,pip,libtorch, source): - Build command you used (if compiling from source):
- Are you using local sources or building from archives:
- Python version:
- CUDA version:
- GPU models and configuration:
- Any other relevant information: