Open ifeherva opened 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?
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?