AutoAlbument's FasterAutoAugmentSearcher is yielding an import error cannot import name 'Batch' from 'torchtext.data' when searching for a policy. This bug recently occurred in PyTorch Lightning in general, and it is fixed by importing Lightning from github: pip install git+https://github.com/PyTorchLightning/pytorch-lightning instead of pip install pytorch-lightning. I suspect that the version of Lightning in the implementation of FasterAutoAugmentSearcher is must be upgraded.
Here's an example of the problem using the CIFAR100 dataset.
Hello,
AutoAlbument's
FasterAutoAugmentSearcher
is yielding an import errorcannot import name 'Batch' from 'torchtext.data'
when searching for a policy. This bug recently occurred in PyTorch Lightning in general, and it is fixed by importing Lightning from github:pip install git+https://github.com/PyTorchLightning/pytorch-lightning
instead ofpip install pytorch-lightning
. I suspect that the version of Lightning in the implementation ofFasterAutoAugmentSearcher
is must be upgraded.Here's an example of the problem using the CIFAR100 dataset.
AutoAlbument Search
dataset.py
Search Output