pytorch
4996e3ac - TensorRT 6.0 support and PyTorch->ONNX->TRT6 unit test (#26426)

Commit
5 years ago
TensorRT 6.0 support and PyTorch->ONNX->TRT6 unit test (#26426) Summary: This PR makes Caffe2 compatible with TensorRT 6. To make sure it works well, new unit test is added. This test checks PyTorch->ONNX->TRT6 inference flow for all classification models from TorhchVision Zoo. Note on CMake changes: it has to be done in order to import onnx-tensorrt project. See https://github.com/pytorch/pytorch/issues/18524 for details. Pull Request resolved: https://github.com/pytorch/pytorch/pull/26426 Reviewed By: hl475 Differential Revision: D17495965 Pulled By: houseroad fbshipit-source-id: 3e8dbe8943f5a28a51368fd5686c8d6e86e7f693
Author
Parents
Loading