Shape-Guided
Shape-Guided copied to clipboard
Can not increase the batchsize
def get_data_loader(split, class_name, img_size, datasets_path, grid_path, shuffle=False):
if split in ['train', 'validation']:
dataset = MVTec3DTrain(split=split, class_name=class_name, img_size=img_size, grid_path=grid_path)
elif split in ['test']:
dataset = MVTec3DTest(class_name=class_name, img_size=img_size, dataset_path=datasets_path, grid_path=grid_path)
data_loader = DataLoader(dataset=dataset, batch_size=1, shuffle=shuffle, num_workers=1, drop_last=False, pin_memory=True)
return data_loader
when I tired to increase the batchsize, error will happen. it says the list must be the same length. Why you use batch_size=1 here?