The following figure presents a pictorial overview of the module
(source).
In this repository, we implement the TokenLearner module and demonstrate its performance with a ViT tiny and the CIFAR-10 dataset.
pip install -r requirements.txt
python train_classifier.py