def __init__(self, my_loader):
    self.my_loader = my_loader
    self.loader_iters = [iter(loader) for loader in self.my_loader.loaders]
    self.loader_iter_sizes = [len(i) for i in self.loader_iters]
    self.max_count = len(self.my_loader)
    self.batch_count = 0
def __init__(self, my_loader):
    self.my_loader = my_loader
    self.loader_iters = [iter(loader) for loader in self.my_loader.loaders]
    self.loader_iter_sizes = [len(i) for i in self.loader_iters]
    self.max_count = len(self.my_loader)
    self.batch_count = 0