Ejemplos de sigmoid_focalloss_forward en Python

Ejemplo n.º 1

    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

Ejemplo n.º 2

Archivo: sigmoid_focal_loss.py Proyecto: huyq/maskrcnn-benchmark

    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