mixmo

Official MixMo implementation in pytorch

Modules

mixmo.augmentations

Data augmentation functions

mixmo.core

Loss, metrics and schedulers

mixmo.learners

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

mixmo.loaders

Dataset wrappers and processing

mixmo.networks

Networks used in the main paper

mixmo.utils

General utility functions