Code for DualViT (ICPR 2024)
- Code, Training and evaluation of DualViT models on CIFAR-10, CIFAR-100, and ImageNet-1k datasets
- Clone the repository:
git clone https://github.com/Dutta-SD/DualViT
- Create a new virtual environment and activate it:
python -m venv env
source env/bin/activate # On Windows, use `env\Scripts\activate`
- Install the required packages:
pip install -r requirements.txt
This project is licensed under the MIT License.
This project utilizes the following third-party libraries:
We would like to thank the authors of the original Vision Transformer paper and the developers of the PyTorch, PyTorch Lightning, and Transformers libraries for their contributions to the field of deep learning and computer vision.