Commit Graph

2 Commits (cd43c4440ef5aec02c3eb242b6f0cade225bb0e2)

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