Commit Graph

12 Commits (100830e5efc06f259b347d55edda90f3e75c14f1)

Author SHA1 Message Date
Darren 37ccd3683d
unify gen_wts.py and inference.cpp dummy test value. (#457)
Reviewed by: @L1aoXingyu
2021-04-19 10:23:15 +08:00
Darren 8276ccf4fd
[v005] set INT8 calibrate set via cmake (#459)
Reviewed by: @L1aoXingyu
2021-04-19 10:22:48 +08:00
Darren e0ad8c70bc
update dockerfile in fastrt (#437)
Reviewed by: @L1aoXingyu
2021-04-12 15:06:13 +08:00
慕湮 fc67350e99
Add python interface by pybind11 and Int8 mode
Reviewed by: @TCHeish
2021-04-12 15:05:21 +08:00
Darren d7c1294d9e
fastrt patch update
Summary:
move div255 to gpu
add read/write numpy ndarray, which will make the comparison between torch and trt results more easily.

Reviewed By: l1aoxingyu
2021-03-10 10:53:18 +08:00
darrenhsieh b9bda486f0 add trt api div255 2021-02-27 16:45:29 +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
darrenhsieh 5f7d3d586e refactor: model.h 2021-01-31 18:12:35 +08:00
darrenhsieh ebc375e51e fix: cmake bug 2021-01-31 18:08:25 +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