fast-reid/fastreid/modeling/losses
liaoxingyu2 a00e50d37f fix triplet ddp training
Summary: fixup precision alignment when triplet loss with ddp
2020-11-06 11:01:10 +08:00
..
__init__.py fix typro bce loss 2020-09-18 10:39:10 +08:00
circle_loss.py fix pair-wise circle loss 2020-09-09 15:28:52 +08:00
cross_entroy_loss.py refactor model arch 2020-09-01 16:14:45 +08:00
focal_loss.py fix bug in mgn #272 2020-09-17 18:17:53 +08:00
smooth_ap.py onnx/trt support 2020-07-29 17:43:39 +08:00
triplet_loss.py fix triplet ddp training 2020-11-06 11:01:10 +08:00
utils.py feat: support amp training 2020-09-02 18:03:12 +08:00