search

traveller59 / torch2trt

convert torch module to tensorrt network or tvm function
MIT License
89 stars 19 forks source link

error with basic example #2

Closed muzi2045 closed 5 years ago

muzi2045 commented 5 years ago

test with basic example code, an error occured:

Traceback (most recent call last):
  File "trt_test.py", line 27, in <module>
    trt_mode_out = graph_pth(img, verbose=True)
  File "/home/aisimba/.local/lib/python3.6/site-packages/torch2trt-1.0.0-py3.6.egg/torch2trt/core.py", line 665, in __call__
  File "/home/aisimba/.local/lib/python3.6/site-packages/torch2trt-1.0.0-py3.6.egg/torch2trt/core.py", line 497, in resolve_graph
  File "/home/aisimba/.local/lib/python3.6/site-packages/torch2trt-1.0.0-py3.6.egg/torch2trt/core.py", line 492, in resolve_graph
  File "/home/aisimba/.local/lib/python3.6/site-packages/torch2trt-1.0.0-py3.6.egg/torch2trt/handlers/constant.py", line 46, in prim_int
TypeError: int() argument must be a string, a bytes-like object or a number, not 'NoneType'