def loaddata(self, train_dir, batch_size, shuffle,is_train=True):
image_datasets = roadDataset(train_dir,is_train=is_train)
# dataset_loaders = torch.utils.data.DataLoader(image_datasets, batch_size=batch_size, shuffle=False, num_workers=self.num_workers, pin_memory=True)
dataset_loaders = torch.utils.data.DataLoader(image_datasets, batch_size=4, shuffle=False, num_workers=4)
data_set_sizes = len(image_datasets)
return dataset_loaders, data_set_sizes
def loaddata(self, train_dir, batch_size, shuffle, is_train=True):
image_datasets = roadDataset(train_dir, is_train=is_train)
train_sampler = DistributedSampler(image_datasets)
dataset_loaders = torch.utils.data.DataLoader(
image_datasets,
batch_size=batch_size,
sampler=train_sampler,
shuffle=shuffle,
num_workers=self.num_workers,
pin_memory=True)
data_set_sizes = len(image_datasets)
return dataset_loaders, data_set_sizes