def forward(ctx, logits, targets, num_classes, gamma, alpha): ctx.save_for_backward(logits, targets) ctx.num_classes = num_classes ctx.gamma = gamma ctx.alpha = alpha losses = _C.sigmoid_focalloss_forward(logits, targets, num_classes, gamma, alpha) return losses
def forward(ctx, logits, targets, gamma, alpha, dtype): if dtype == 'float16': logits = logits.float() ctx.save_for_backward(logits, targets) num_classes = logits.shape[1] ctx.num_classes = num_classes ctx.gamma = gamma ctx.alpha = alpha ctx.dtype = dtype losses = _C.sigmoid_focalloss_forward(logits, targets, num_classes, gamma, alpha) return losses