> 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
@sonack You can email me your wechat id.