torch_em.classification

Dataset, DataLoader and Trainer implementations for image classification tasks.

1"""Dataset, DataLoader and Trainer implementations for image classification tasks.
2"""
3from .classification import default_classification_trainer, default_classification_loader
4
5from .classification_logger import ClassificationLogger
6from .classification_trainer import ClassificationTrainer