12 Commits (master)

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