How to add attention module to train

[shaun881017]

how can i add attention like CBAM to train to see if the result is better?

[achel-x]

I wanna know too. In home page of this repo, I saw the CBAM module is supported in June 17th this year. But I don't know where it is and how to use it in train and inference.