mixmo.core.loss

Base loss definitions

Functions

get_loss(config_loss[, device, config_args])

Construct loss object, wrapped if there are multiple losses

Classes

AbstractLoss(config_args, device[, config_loss])

Base loss class defining printing and logging utilies

SoftCrossEntropyLoss(config_args, device[, …])

Soft CrossEntropy loss that specifies the proper forward function for AbstractLoss

WrapperLoss(config_loss, config_args, device)

Wrapper around the multiple losses.