fast-reid/fastreid/modeling/losses
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 onnx/trt support 2020-07-29 17:43:39 +08:00
circle_loss.py onnx/trt support 2020-07-29 17:43:39 +08:00
cross_entroy_loss.py onnx/trt support 2020-07-29 17:43:39 +08:00
focal_loss.py finish v0.2 ddp training 2020-07-06 16:57:43 +08:00
smooth_ap.py onnx/trt support 2020-07-29 17:43:39 +08:00
triplet_loss.py onnx/trt support 2020-07-29 17:43:39 +08:00
utils.py onnx/trt support 2020-07-29 17:43:39 +08:00