return torch._VF.split_with_sizes(self, split_size, dim)
IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1)
原因yolo forward之前改了为了ncnn能导出 后来没看久了忘了 mark一下
return torch._VF.split_with_sizes(self, split_size, dim)
IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1)
原因yolo forward之前改了为了ncnn能导出 后来没看久了忘了 mark一下