Augmentation error - TrivialAugmentWide

vision
1

Hi

I am trying to utilize the PyTorch augmentation TrivialAugmentWide() but I get the error message.

cannot import name ‘TrivialAugmentWide’ from ‘torchvision.transforms.autoaugment’

Any help is appreciated.

Az

2

You might need to update torchvision as this transformation was added ~2 months ago.