torch2trt: Can not convert interpolate

> python3 -m torch2trt.test --name=interpolate
Traceback (most recent call last):
  File "/usr/local/lib/python3.7/runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "/usr/local/lib/python3.7/runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "/usr/local/lib/python3.7/site-packages/torch2trt-0.0.0-py3.7.egg/torch2trt/test.py", line 100, in <module>
    max_error, fps, fps_trt, ms, ms_trt = run(test)
  File "/usr/local/lib/python3.7/site-packages/torch2trt-0.0.0-py3.7.egg/torch2trt/test.py", line 25, in run
    module_trt = torch2trt(module, inputs, **self.torch2trt_kwargs)
  File "/usr/local/lib/python3.7/site-packages/torch2trt-0.0.0-py3.7.egg/torch2trt/torch2trt.py", line 252, in torch2trt
    outputs = module(*inputs)
  File "/usr/local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 539, in __call__
    result = self.forward(*input, **kwargs)
  File "/usr/local/lib/python3.7/site-packages/torch2trt-0.0.0-py3.7.egg/torch2trt/converters/interpolate/interpolate.py", line 49, in forward
    return F.interpolate(x, self.size, mode=self.mode, align_corners=self.align_corners)
  File "/usr/local/lib/python3.7/site-packages/torch2trt-0.0.0-py3.7.egg/torch2trt/torch2trt.py", line 97, in wrapper
    converter(ctx)
  File "/usr/local/lib/python3.7/site-packages/torch2trt-0.0.0-py3.7.egg/torch2trt/converters/interpolate/interpolate.py", line 34, in convert_interpolate
    plugin = get_interpolate_plugin(size=size, mode=mode, align_corners=align_corners)
  File "/usr/local/lib/python3.7/site-packages/torch2trt-0.0.0-py3.7.egg/torch2trt/converters/interpolate/interpolate.py", line 11, in get_interpolate_plugin
    creator = [c for c in registry.plugin_creator_list if c.name == PLUGIN_NAME and c.plugin_namespace == 'torch2trt'][0]
IndexError: list index out of range

About this issue

  • Original URL
  • State: closed
  • Created 5 years ago
  • Comments: 26 (12 by maintainers)

Most upvoted comments

@sonack You can email me your wechat id.