def __iter__(self):
    iterator = MtUcertaintyIterator(self)
    for l in self.loaders:
        if isinstance(l.sampler, DistributedSampler):
            l.sampler.set_epoch(self.current_epoch)
    self.current_epoch += 1
    return iterator
def __iter__(self):
    iterator = MtUcertaintyIterator(self)
    for l in self.loaders:
        if isinstance(l.sampler, DistributedSampler):
            l.sampler.set_epoch(self.current_epoch)
    self.current_epoch += 1
    return iterator