关于torchmetrics的版本问题_torchmetrics与pytorch版本对应

作者：神奇cpp | 2024-08-13 09:47:56

踩

torchmetrics与pytorch版本对应

torchmetrics的版本一定要跟 PyTorch Lightning的版本对应，否则可能会出现数据溢出的问题，比如在计算AUROC时会把形状为[22,2]的pred在计算时自动转成[44,1]的大小，而target仍然为[22,1],这时就会出现数据溢出的问题。

具体报错如下，使用的torchmetrics==0.10.1，pytorch-lightning==1.2.3

根据这篇知乎，将torchmetrics==0.7.0就没再报错了

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/神奇cpp/article/detail/974273