fast-reid/fastreid/modeling/heads
liaoxingyu 16655448c2 onnx/trt support
Summary: change model pretrain mode and support onnx/TensorRT export
2020-07-29 17:43:39 +08:00
..
__init__.py fix(heads): fix bug in reduce head 2020-05-23 10:41:13 +08:00
bnneck_head.py onnx/trt support 2020-07-29 17:43:39 +08:00
build.py refactor(heads): move num_classes out from heads 2020-04-29 21:29:48 +08:00
linear_head.py onnx/trt support 2020-07-29 17:43:39 +08:00
reduction_head.py onnx/trt support 2020-07-29 17:43:39 +08:00