警告信息:
UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument
nn.Softmax()函数已经被弃用了,虽然程序还是可以运行成功,但是这个做法不被pytorch所赞成。
这个写法在早期的pytorch版本是没有警告的,现在因为其他考虑,要加上有指明dim参数。
改成nn.Softmax(dim=1) 即可
本文讨论了PyTorch中Softmax函数的使用警告,解释了为什么默认维度选择已被弃用,并提供了更新的调用方式以避免警告信息。建议将nn.Softmax()更改为nn.Softmax(dim=1),以明确指定维度。
警告信息:
UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument
nn.Softmax()函数已经被弃用了,虽然程序还是可以运行成功,但是这个做法不被pytorch所赞成。
这个写法在早期的pytorch版本是没有警告的,现在因为其他考虑,要加上有指明dim参数。
改成nn.Softmax(dim=1) 即可
4104
2053
1万+
3427
2万+

被折叠的 条评论
为什么被折叠?