Commit Graph

2 Commits (1c08a2136e1fdd9579cfa3bec2dd19f9cf1c1c10)

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