How to use the autogluon.loss.autogluon_losses function in autogluon

To help you get started, we’ve selected a few autogluon examples, based on popular ways it is used in public projects.

Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.

github awslabs / autogluon / autogluon / task / image_classification / losses.py View on Github external
@autogluon_losses
def MixSoftmaxCrossEntropyLoss(**kwargs):
    pass
github awslabs / autogluon / autogluon / task / image_classification / losses.py View on Github external
@autogluon_losses
def SoftmaxCrossEntropyLoss(**kwargs):
    pass
github awslabs / autogluon / autogluon / task / image_classification / losses.py View on Github external
@autogluon_losses
def SigmoidBinaryCrossEntropyLoss(**kwargs):
    pass
github awslabs / autogluon / autogluon / task / image_classification / losses.py View on Github external
@autogluon_losses
def get_loss(name, **kwargs):
    """Returns a loss with search space by name

    Args:
        name : str
            Name of the loss.
    """
    if name not in losses and name.lower() not in losses:
        err_str = '"%s" is not among the following loss list:\n\t' % (name)
        err_str += '%s' % ('\n\t'.join(sorted(losses.keys())))
        raise ValueError(err_str)
    loss = Loss(name)
    return loss
github awslabs / autogluon / autogluon / task / image_classification / losses.py View on Github external
@autogluon_losses
def DistillationSoftmaxCrossEntropyLoss(**kwargs):
    pass