Awesome
Vision Mamba Distillation for Low-resolution Fine-grained Image Classification
Configuration
Python 3 Pytorch 2.1
Training
./ViMD_train_teacher.sh
Please modify the model path in ViMD_train_teacher.sh and ViMD_train_teacher.py
./ViMD_train_student.sh
Please modify the model path in ViMD_train_student.sh and ViMD_train_student.py
Testing
./ViMD_test.sh