def get_data_loader(name, train=True): """Get data loader by name.""" if name == "MNIST": return get_mnist(train) elif name == "USPS": return get_usps(train) elif name == "SVHN": return get_svhn(train)
def get_data_loader(name, train=True, get_dataset=False): """Get data loader by name.""" if name == "MNIST": return get_mnist(train, get_dataset) elif name == "MNIST-M": return get_mnist_m(train, get_dataset) elif name == "SVHN": return get_svhn(train, get_dataset) elif name == "USPS": return get_usps(train, get_dataset)
def get_data_loader(name, dataset_root, batch_size, train=True): """Get data loader by name.""" if name == "mnist": return get_mnist(dataset_root, batch_size, train) elif name == "mnistm": return get_mnistm(dataset_root, batch_size, train) elif name == "svhn": return get_svhn(dataset_root, batch_size, train) elif name == "amazon31": return get_office(dataset_root, batch_size, 'amazon') elif name == "webcam31": return get_office(dataset_root, batch_size, 'webcam') elif name == "webcam10": return get_officecaltech(dataset_root, batch_size, 'webcam')
def get_data_loader(name, train=True): """Get data loader by name.""" if name == "MNIST": dataset = get_mnist(train) elif name == "MNISTM": dataset = get_mnistm(train) elif name == "USPS": dataset = get_usps(train) elif name == "SVHN": dataset = get_svhn(train) if train: data_loader = torch.utils.data.DataLoader(dataset=dataset, batch_size=params.batch_size, shuffle=True) else: data_loader = torch.utils.data.DataLoader(dataset=dataset, batch_size=1, shuffle=False) return data_loader
def get_data_loader(name, train=True, dataset=None, sample=False): """Get data loader by name.""" if name == "MNIST": return get_mnist(train, sample) elif name == "MNIST-M": return get_mnist_m(train) elif name == "SVHN": return get_svhn(train) elif name == "USPS": return get_usps(train, sample) elif name == "SYNTH": return get_synth(train) elif name == "SYNTHSIGN": return get_synthsign(train) elif name == "GTSRB": return get_gtsrb(train) elif name == "STL": return get_stl(train) elif name == "CIFA": return get_cifa(train) elif name == "PIE27": return get_pie27(train) elif name == "PIE05": return get_pie05(train) elif name == "PIE09": return get_pie09(train) elif name == "PIE37": return get_pie37(train) elif name == "PIE25": return get_pie25(train) elif name == "PIE02": return get_pie02(train) elif name == "taskcv_S": return get_taskcv_s(train) elif name == "taskcv_T": return get_taskcv_t(train) elif name == "OFFICE": return get_office(dataset)
def get_data_loader(name, train=True, split='train'): """Get data loader by name.""" if name == "svhn": return get_svhn(split) elif name == "mnist": return get_mnist(train)