Commit Graph

2 Commits

Author SHA1 Message Date
irvingzhang0512
501160d262
Support TSM-R50 C++ API (#500)
* C++ API for TSM-R50

* fix bugs
2021-04-23 14:57:01 +08:00
irvingzhang0512
8cfc8ee03f
Support TSM-R50 Python API (#488)
* add tensorrt temporal shift module and related pytorch implementations

* add .gitignore and getn weights script.

* rename get_wts.py script

* Add tsm-r50 demo.

* update readme

* remove useless codes

* update readme

* update readme

* remote video and .gitignore, update tutorial

* update readme and tutorial

* fix a few bugs and test on tensorrt 5.1

* update readme
2021-04-18 16:30:25 +08:00