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