search

grimoire / torch2trt_dynamic

A pytorch to tensorrt convert with dynamic shape support
MIT License
255 stars 34 forks source link

Import ERROR #38

Open jg0014 opened 3 months ago

jg0014 commented 3 months ago

tensorrt 8.4.1.5

import torch2trt_dynamic Traceback (most recent call last): File "", line 1, in File "/home_orin/anaconda3/envs/track2/lib/python3.8/site-packages/torch2trt_dynamic/init.py", line 3, in from .converters import * # noqa: F401,F403 File "/home_orin/anaconda3/envs/track2/lib/python3.8/site-packages/torch2trt_dynamic/converters/init.py", line 19, in from . import grid_sample # noqa: F401 File "/home_orin/anaconda3/envs/track2/lib/python3.8/site-packages/torch2trt_dynamic/converters/grid_sample.py", line 8, in bicubic=trt.ResizeMode.CUBIC) AttributeError: type object 'tensorrt.tensorrt.ResizeMode' has no attribute 'CUBIC'

grimoire commented 3 months ago

Update tensorrt version or just remove bicubic resize model in torch2trt_dynamic/converters/grid_sample.py.