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

Rebuild DPU kernel #220

Open lloo099 opened 2 years ago

lloo099 commented 2 years ago

Hi, thanks for your great work and sharing. I encounter some problems when I compiled the mobilenet_v2_tfslim.py after initialized to tensorflow-slim submodule (git submodule update --init --recursive). And build online_model fold succesfully.

However, the error was happened in this line code


  File "mobilenet_v2_tfslim.py", line 235, in <module>
    if IMPORT_PYTORCH and not import_pytorch_weights():
  File "mobilenet_v2_tfslim.py", line 119, in import_pytorch_weights
    import mobilenet_v2_tsm_v2 as tsm_online
ModuleNotFoundError: No module named 'mobilenet_v2_tsm_v2'