def __len__(self):
    loader_len = [len(loader) for loader in self.loaders]
    if self.args.uniform_mt_sampling:
        return int((((self.args.percent_of_max_data_size * max(loader_len)) * len(self.loaders)) / self.args.train_batch_size))
    elif self.args.uncert_batch:
        return int(((max(loader_len) * len(self.loaders)) * self.args.percent_of_max_data_size))
    else:
        return sum(loader_len)
def __len__(self):
    loader_len = [len(loader) for loader in self.loaders]
    if self.args.uniform_mt_sampling:
        return int((((self.args.percent_of_max_data_size * max(loader_len)) * len(self.loaders)) / self.args.train_batch_size))
    elif self.args.uncert_batch:
        return int(((max(loader_len) * len(self.loaders)) * self.args.percent_of_max_data_size))
    else:
        return sum(loader_len)