Commit Graph

2 Commits (7f7dfccf20347eb9f0600b15a6472c32f1c34c4b)

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