def __init__(self, loaders, datasets, loader_args):
    self.loaders = loaders
    self.dataset = datasets
    self.args = loader_args
    self.current_epoch = 0
def __init__(self, loaders, datasets, loader_args):
    self.loaders = loaders
    self.dataset = datasets
    self.args = loader_args
    self.current_epoch = 0