RuntimeError: probability tensor contains either `inf`, `nan` or element ＜ 0
这个问题卡了有一天的时间，后来，在对于torch的每个版本进行测试后，发现torch 2.2会出现这个问题，将版本降到torch 2.1.0就可以解决。
·
这个问题卡了有一天的时间，后来，在对于torch的每个版本进行测试后，发现torch 2.2会出现这个问题，将版本降到torch 2.1.0就可以解决。
pip install --upgrade torch==2.1.2
技术共进，成长同行——讯飞AI开发者社区
更多推荐
5
0- 0
已为社区贡献5条内容
所有评论(0)