mixmo.learners.model_wrapper

Utility definitions to wrap a model with losses, metrics and logs

Functions

get_predictions(logits)

Convert logits into softmax predictions

Classes

ModelWrapper(config, config_args, device)

Augment a model with losses, metrics, internal logs and other things