def get_dataflow(config):
    if (idist.get_rank() > 0):
        idist.barrier()
    (train_dataset, test_dataset) = utils.get_train_test_datasets(config['data_path'])
    if (idist.get_rank() == 0):
        idist.barrier()
    train_loader = idist.auto_dataloader(train_dataset, batch_size=config['batch_size'], num_workers=config['num_workers'], shuffle=True, drop_last=True)
    test_loader = idist.auto_dataloader(test_dataset, batch_size=(2 * config['batch_size']), num_workers=config['num_workers'], shuffle=False)
    return (train_loader, test_loader)
def get_dataflow(config):
    if (idist.get_rank() > 0):
        idist.barrier()
    (train_dataset, test_dataset) = utils.get_train_test_datasets(config['data_path'])
    if (idist.get_rank() == 0):
        idist.barrier()
    train_loader = idist.auto_dataloader(train_dataset, batch_size=config['batch_size'], num_workers=config['num_workers'], shuffle=True, drop_last=True)
    test_loader = idist.auto_dataloader(test_dataset, batch_size=(2 * config['batch_size']), num_workers=config['num_workers'], shuffle=False)
    return (train_loader, test_loader)