Problem with the torch_Dataloader_OASIS function, specifically with the drop_last=True parameter

vision
1

I am working on Unsupervised Medical Images Segmentation unable to set the path according to database. It seems like that Dataloader function is not working properly. Below is my code.

#The path accordingly
OASIS_img_dir = ‘i set the path according to my directory’
OASIS_seg_dir = ‘i set the path according to my directory’
resized_shape = (80, 96, 112

train_OASIS = torch_Dataloader_OASIS(OASIS_img_dir, OASIS_seg_dir, ‘train’, resized_shape, 1, None)
test_OASIS = torch_Dataloader_OASIS(OASIS_img_dir, OASIS_seg_dir, ‘test’, resized_shape, 1, None))

Got this error:
line 22, in
train_OASIS = torch_Dataloader_OASIS(OASIS_img_dir, OASIS_seg_dir, ‘train’, resized_shape, 1, None)