torch2trt
torch2trt copied to clipboard
Is there a plan to support python 3.10?
Model export is failing on:
File "/venv/lib/python3.10/site-packages/torch2trt-0.4.0-py3.10.egg/torch2trt/torch2trt.py", line 310, in wrapper
converter["converter"](ctx)
File "/venv/lib/python3.10/site-packages/torch2trt-0.4.0-py3.10.egg/torch2trt/converters/interpolate.py", line 81, in convert_interpolate_trt7
if not isinstance(scales, collections.Sequence):
AttributeError: module 'collections' has no attribute 'Sequence'
Because it has been dropped from the collections module in python 3.10
I added a pull request, fixing this issue. Please check it out. It was a simple fix. https://github.com/NVIDIA-AI-IOT/torch2trt/pull/884