torch_em.self_training

 1"""Training procedures for unsupervised self training of neural networks.
 2"""
 3
 4from .logger import SelfTrainingTensorboardLogger
 5from .loss import DefaultSelfTrainingLoss, DefaultSelfTrainingLossAndMetric, ProbabilisticUNetLoss, \
 6    ProbabilisticUNetLossAndMetric
 7from .mean_teacher import MeanTeacherTrainer
 8from .fix_match import FixMatchTrainer
 9from .pseudo_labeling import DefaultPseudoLabeler, ProbabilisticPseudoLabeler
10from .probabilistic_unet_trainer import ProbabilisticUNetTrainer, DummyLoss