search

leaderj1001 / Stand-Alone-Self-Attention

Implementing Stand-Alone Self-Attention in Vision Models using Pytorch
MIT License
456 stars 83 forks source link

Does not work when out_channels is not even #12

Open ifeherva opened 4 years ago

ifeherva commented 4 years ago

In my case out_channels is not an even number so v_out_h, v_out_w = v_out.split(self.out_channels // 2, dim=1) fails, any idea how to fix it?