Notice
Recent Posts
Recent Comments
Link
«   2025/02   »
일 월 화 수 목 금 토
1
2 3 4 5 6 7 8
9 10 11 12 13 14 15
16 17 18 19 20 21 22
23 24 25 26 27 28
Tags
more
Archives
Today
Total
관리 메뉴

Zorba blog

[Error] ImportError: cannot import name 'get_num_classes' from 'torchmetrics.utilities.data' 본문

카테고리 없음

[Error] ImportError: cannot import name 'get_num_classes' from 'torchmetrics.utilities.data'

Zorba blog 2022. 5. 23. 15:31

Kobart를 사용하여 전이학습을 진행하려고 하는 도중

ImportError: cannot import name 'get_num_classes' from 'torchmetrics.utilities.data'

torchmetrics 에서 에러 발생.

 

1. 에러 원인

Traceback (most recent call last):
  File "train.py", line 6, in <module>
    import pytorch_lightning as pl
  File "/anaconda3/envs/kobart/lib/python3.7/site-packages/pytorch_lightning/__init__.py", line 20, in <module>
    from pytorch_lightning import metrics  # noqa: E402
  File "/anaconda3/envs/kobart/lib/python3.7/site-packages/pytorch_lightning/metrics/__init__.py", line 15, in <module>
    from pytorch_lightning.metrics.classification import (  # noqa: F401
  File "/anaconda3/envs/kobart/lib/python3.7/site-packages/pytorch_lightning/metrics/classification/__init__.py", line 14, in <module>
    from pytorch_lightning.metrics.classification.accuracy import Accuracy  # noqa: F401
  File "/anaconda3/envs/kobart/lib/python3.7/site-packages/pytorch_lightning/metrics/classification/accuracy.py", line 18, in <module>
    from pytorch_lightning.metrics.utils import deprecated_metrics
  File "/anaconda3/envs/kobart/lib/python3.7/site-packages/pytorch_lightning/metrics/utils.py", line 22, in <module>
    from torchmetrics.utilities.data import get_num_classes as _get_num_classes
ImportError: cannot import name 'get_num_classes' from 'torchmetrics.utilities.data' (/anaconda3/envs/kobart/lib/python3.7/site-packages/torchmetrics/utilities/data.py)

torchmetrics의 버전 문제로 인해 에러가 발생.

 

2. 해결 방법

$ pip install torchmetrics==0.6.0

 

버전을 다운그레이드 하면 된다.

 

3. 참고링크

https://github.com/NVIDIA/DeepLearningExamples/issues/1113

Comments