About PSM module

I attempted to verify the functionality of the PSM module and noticed that incorporating only the PSM module into ViT didn't seem to enhance performance. Could you please let me know if there is a transfg model weights trained on CUB?