Commit Graph

3 Commits (0617e3eeb720dfd267333e2a5149260eac7712ac)

Author SHA1 Message Date
darrenhsieh 0617e3eeb7 refactor: embedding_head
* add abstract poolinglayer

* add poolinglayer factory
2021-02-13 21:28:29 +08:00
darrenhsieh 159494e4a4 refactor: model class
* add Module as base class

* add head type

* replace template func args with Module ptr

* add Module factory
2021-02-13 21:14:14 +08:00
darrenhsieh dc5f1924dc feat: support reid model with trt C++APIs
Summary: support reid model with tensorrt network definition APIs
2021-01-30 15:40:05 +08:00