🐛 [Bug] An wrong function call for torch._C._jit_to_tensorrt
on spec
in tutorial code.
#286
Labels
Projects
torch._C._jit_to_tensorrt
on spec
in tutorial code.
#286
Bug Description
This is a bug in a tutorial below.
To compile a model,
torch._C._jit_to_tensorrt()
is called with two parameters,script_model._c
andspec
.But, a following error happened.
This error looks like caused by wrong parameter. In test code (https://github.com/NVIDIA/TRTorch/blob/b93627ecc7bb7123d2e32aff9d762dd0e6bc3166/tests/py/test_to_backend_api.py), this API is called with a different parameter like below. It looks like making additional
dict
withforward
key is necessary.To Reproduce
Steps to reproduce the behavior:
nvcr.io/nvidia/tensorrt:20.03-py3
.pip install https://github.com/NVIDIA/TRTorch/releases/download/v0.1.0/trtorch-0.1.0-cp36-cp36m-linux_x86_64.whl torchvision==0.7.0
My repro code is below. Note that I modified the tutorial code to use FP32 for test.
Expected behavior
No exception.
Environment
conda
,pip
,libtorch
, source): pipAdditional context
N/A
The text was updated successfully, but these errors were encountered: