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 .
image784×581 63.5 KB
Here’s the link to my notebook 1 where I am getting  the error .
Please help me in suggesting the syntax , so that I can correct the error present .