Commit Graph

2 Commits (ed292695c577c85e730931daf948b8e4dd0236dc)

Author SHA1 Message Date
FlyingQianMM 944c3165ec
fix type error of std::pow in sigmoid_focal_loss_op.cu and sigmoid_focal_loss_op.h (#18152)
6 years ago
FlyingQianMM 0aee1f0074
add sigmoid focal loss operator for supporting retinanet (#17895)
6 years ago