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)
