警告信息：
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) 即可
Implicit dimension choice for softmax has been deprecated. yolo
最新推荐文章于 2024-11-29 16:18:33 发布
本文讨论了PyTorch中Softmax函数的使用警告，解释了为什么默认维度选择已被弃用，并提供了更新的调用方式以避免警告信息。建议将nn.Softmax()更改为nn.Softmax(dim=1)，以明确指定维度。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理
扫一扫
1589
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
