Custom Image dataset working only for validation data

vision
1

I have created a custom dataset class for my training and validation dataset . Among which the torch transform with albumentations library is working only for validation data and not for my training data .It is giving back the below error .

image
image784×581 63.5 KB

Here’s the link to my notebook where I am getting the error .
Please help me in suggesting the syntax , so that I can correct the error present .

2

For training you are using a transform ‘Cutout’ from torchtoolbox library which I believe does not have ‘always_apply’, why don’t you use ‘Cutout’ from albumentations itself?

3

thanks …but I already resolved the issue