torch_em.self_training

 1"""Training procedures for unsupervised self training of neural networks.
 2"""
 3
 4from .logger import SelfTrainingTensorboardLogger, UniMatchv2TensorboardLogger
 5from .loss import DefaultSelfTrainingLoss, DefaultSelfTrainingLossAndMetric, ProbabilisticUNetLoss, \
 6    ProbabilisticUNetLossAndMetric, SelfTrainingLossWithInvertibleAugmentations, \
 7    SelfTrainingLossAndMetricWithInvertibleAugmentations, \
 8    UniMatchv2Loss, UniMatchv2LossAndMetric
 9from .mean_teacher import MeanTeacherTrainer, MeanTeacherTrainerWithInvertibleAugmentations
10from .fix_match import FixMatchTrainer, FixMatchTrainerWithInvertibleAugmentations
11from .pseudo_labeling import DefaultPseudoLabeler, ProbabilisticPseudoLabeler, ScheduledPseudoLabeler
12from .probabilistic_unet_trainer import ProbabilisticUNetTrainer, DummyLoss
13from .uni_match_v2 import UniMatchv2Trainer