UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument.
input = module(input)
这个警告的原因是softmax()函数已经被弃用了,虽然程序还是可以运行成功,但是这个做法不被pytorch所赞成。这个写法在早期的pytorch版本是没有警告的,现在因为其他考虑,要加上有指明dim参数。
nn.Softmax()
改成
nn.Softmax(dim=1) #即在第1维上概率sum=1
dim=某个数字,这个数字可以根据自己的需要更改。
————————————————
版权声明:本文为优快云博主「York1996」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.youkuaiyun.com/york1996/article/details/84191229
本文详细解析了PyTorch中Softmax函数的使用方法,针对未指定dim参数的警告进行了说明,并提供了正确的代码修改建议。通过明确指定dim参数,可以避免未来版本可能的兼容性问题。
1597





