Commit Graph

4 Commits (fc67350e992bc20903cb2e2321ce6e1db92d06e3)

Author SHA1 Message Date
慕湮 fc67350e99
Add python interface by pybind11 and Int8 mode
Reviewed by: @TCHeish
2021-04-12 15:05:21 +08:00
darrenhsieh 69eb044b81 [v004] de-couple fastreid config from Model base-class
fix baseline class ctr bug
2021-02-27 16:40:04 +08:00
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