mixmo.learners

Learner objects that wrap models, optimizers and datasets to train and evaluate

Modules

mixmo.learners.abstract_learner

Base Learner wrapper definitions for logging, training and evaluating models

mixmo.learners.learner

Training and evaluation loop definitions for the Learner objects

mixmo.learners.model_wrapper

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