这篇博客主要讨论了PyTorch更新版本后遇到的两个兼容性问题。第一个问题是softmax函数隐式维度选择被弃用，解决方案是在softmax调用中明确指定dim参数。第二个问题是由于版本差异导致的IndexError，解决办法是调整loss函数的取值方式。通过这些修改，作者解决了版本升级带来的错误。
报错信息：
Implicit dimension choice for soft max has been deprecated. Change the call to include D_out = interp(model_D(F.softmax(pred)))
错误原因：
版本兼容问题。pytorch较高版本会出现此错误，在高版本中，softmax的隐式维度选择已弃用 。需要调用包含dim=X作为参数。
源码：
D_out = interp(model_D(F.softmax(pred)))
修改：
D_out = interp(model_D(F.softmax(pred, dim=1)))
报错信息：
IndexError: too many indices for array.
错误原因：
最低0.47元/天 解锁文章
扫一扫
626
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
