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
更多推荐
所有评论(0)