按照官方介绍:https://pytorch-lightning.readthedocs.io/en/1.4.4/extensions/metrics.html

pytorch_lightning.metric已经被废弃了,里面的一些函数以及包被作者放到了另外一个包里面,需要单独安装,安装pytorch_lightning并不会默认安装它。

这个包就是TorchMetrics,这个玩意也是同一个团队开发的,但是并不属于同一个库。

作者将这个分离出来的原因是,这种metrics的东西可以用在任何有关tensor的模型里面,所以没有必要捆绑在pytorch_lightning中,这样,如果作者不想使用pytorch_lightning,可以单独安装TorchMetrics用在pytorch的模型里面。

这个东西3M的样子,安装方法以及使用见TorchMetrics的安装以及使用。

Logo

科技之力与好奇之心,共建有温度的智能世界

更多推荐