Shape-Guided icon indicating copy to clipboard operation
Shape-Guided copied to clipboard

Can not increase the batchsize

Open limaodaxia opened this issue 11 months ago • 0 comments

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?

limaodaxia avatar Mar 04 '24 01:03 limaodaxia