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?
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?