search

luuuyi / CBAM.PyTorch

Non-official implement of Paper：CBAM: Convolutional Block Attention Module
1.34k stars 285 forks source link

Can pytorch 0.4.0 use this code? #1

Closed gaichengyuan closed 5 years ago

luuuyi commented 5 years ago

@gaichengyuan When I complete this work, my bench is pytorch0.4.1, but I think pytorch0.4.0 can also work. You can try.