Commit Graph

9 Commits (1db050b51f4e722e76e2fcba976255647ae67c0b)

Author SHA1 Message Date
liaoxingyu dbf1604231 fix model deploy problems 2021-05-25 15:55:43 +08:00
Xingyu Liao be0a089e1f
bugfix & merge classification transforms (#448)
Summary: change heads definition in project and config file, merge classification transforms into default transforms
2021-03-30 15:47:14 +08:00
Xingyu Liao cb7a1cb3e1
update deployment toolchain (#428)
Summary:
Remove tiny-tensorrt dependency and rewrite a new tensorrt inference api.
In the new version of trt infer, it can pad the input to fixed batch automatically, so you don't need to worry about dynamic batch size.
2021-03-10 16:48:59 +08:00
liaoxingyu e2a1e14bc3 remove extra info 2021-02-24 14:40:45 +08:00
liaoxingyu 15e1729a27 update fastreid V1.0 2021-01-18 11:36:38 +08:00
liaoxingyu 07872d4cdb fix caffe export bug
Summary: put tensor to gpu device
2020-08-20 16:28:52 +08:00
liaoxingyu 16655448c2 onnx/trt support
Summary: change model pretrain mode and support onnx/TensorRT export
2020-07-29 17:43:39 +08:00
liaoxingyu fec7abc461 finish v0.2 ddp training 2020-07-06 16:57:43 +08:00
liaoxingyu 2f6d999469 add pytorch to caffe converting
Summary: update deployment README and support pytorch to caffe converting for basemodel
2020-06-02 18:11:02 +08:00