请问这个是哪里出了问题，我没改代码

[425her0]

File "/root/pspnet-pytorch-master/utils/callbacks.py", line 135, in get_miou_png pr = F.softmax(pr.permute(1,2,0),dim = -1).cpu().numpy() RuntimeError: number of dims don't match in permute

[425her0]

我把aux_branch设为False就不会有问题了 请问这是什么原因

[bubbliiiing]

额，因为aux_branch模式输出为两个，我没考虑到这个情况，我想想

[bubbliiiing]

修改了