Hello sir ,
Here is a good paper for the attention model with its code impelemented.
I hope it will help you.

  
      
      GitHub
  
  
    

wvinzh/WS_DAN_PyTorch 124

PyTorch Implementation Of WS-DAN(See Better Before Looking Closer: Weakly Supervised Data Augmentation Network for Fine-Grained Visual Classification) - wvinzh/WS_DAN_PyTorch