Commit Graph

2 Commits (a13a4dbc7bfeb2c966df42c5eaaf6d498b6c7eff)

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