如何在pytorch光照中导入TensorMetric？

Question

我要导入 TensorMetric :

from pytorch_lightning.metrics.metric import TensorMetric

程序抛出异常：

ModuleNotFoundError: No module named 'pytorch_lightning.metrics'

我的环境是：

• Python 3.8.13
• 分词器==0.9.2
• 手电筒==1.5.1
• 变形金刚==3.4.0
• pytorch-闪电==0.9.0
• tensorboard==2.2.0

Answer 1

0.9 是一个非常旧的 Lightning 版本，从那时起就有很多 changed/improved。我建议您使用最新版本的 PyTorch Lightning (1.6)。

此外，我们现在将指标作为一个单独的库 TorchMetrics。