### Description The [transformer-based image classification model](https://arxiv.org/abs/2010.11929) is becoming popular. It will be nice to include it in this repo. ### Expected behavior with the suggested feature - similar classification performance as the CNN based models ### Other Comments