search

mit-han-lab / temporal-shift-module

[ICCV 2019] TSM: Temporal Shift Module for Efficient Video Understanding
https://arxiv.org/abs/1811.08383
MIT License
2.05k stars 418 forks source link

TypeError: 'IntImm' object cannot be interpreted as an integer #234

Open SlowMonk opened 4 months ago

SlowMonk commented 4 months ago

Device: JetsonNano OS: ubuntu 22.04

File "/home/surromind/swe/temporal-shift-module/online_demo/main.py", line 389, in main()

  File "/home/temporal-shift-module/online_demo/main.py", line 285, in main
    executor, ctx = get_executor()

  File "/home/temporal-shift-module/online_demo/main.py", line 99, in get_executor
    return torch2executor(torch_module, torch_inputs, target)

  File "/home/temporal-shift-module/online_demo/main.py", line 55, in torch2executor
    graph, tvm_module, params = torch2tvm_module(torch_module, torch_inputs, target)

  File "/home/temporal-shift-module/online_demo/main.py", line 40, in torch2tvm_module
    graph, tvm_module, params = tvm.relay.build(relay_module, target, params=params)

  File "/home/venv/lib/python3.10/site-packages/tvm-0.6.0-py3.10-linux-aarch64.egg/tvm/relay/build_module.py", line 248, in build
    graph_json, mod, params = bld_mod.build(func, target, target_host, params)

  File "/home/venv/lib/python3.10/site-packages/tvm-0.6.0-py3.10-linux-aarch64.egg/tvm/relay/build_module.py", line 107, in build
    target = _update_target(target)

  File "/home/venv/lib/python3.10/site-packages/tvm-0.6.0-py3.10-linux-aarch64.egg/tvm/relay/build_module.py", line 49, in _update_target
    dev_type = _nd.context(dev).device_type

  File "/home/venv/lib/python3.10/site-packages/tvm-0.6.0-py3.10-linux-aarch64.egg/tvm/_ffi/ndarray.py", line 82, in context
    return TVMContext(dev_type, dev_id)

  File "/home/venv/lib/python3.10/site-packages/tvm-0.6.0-py3.10-linux-aarch64.egg/tvm/_ffi/runtime_ctypes.py", line 170, in __init__
    self.device_type = device_type

TypeError: 'IntImm' object cannot be interpreted as an integer