Custom Image dataset working only for validation data

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 .

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 .

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?

thanks …but I already resolved the issue