티스토리 뷰

Computer/이게 왜 안되지?

[이왜안?] ImportError: cannot import name 'LRScheduler' from 'torch.optim.lr_scheduler'

벼랑끝과학자 2025. 8. 7. 19:18

[원인]

torch 버전과 torch를 이용한 training을 최적화해주는 lightening 라이브러리 사이의 업데이트 사이에서 변경점이 서로 충돌하는 것이 원인이라고 생각된다. torch의 lr_scheduler.py파일은 버전에 따라서 class 이름이 LRScheduler인 것도 있지만 _LRScheduler로 언더바가 붙은 버전도 배포된 것으로 생각된다.

왼쪽 : pytorch의 lr_scheduler.pyu 내가 가진 환경은 class 이름이 _LRScheduler로 되어있다. 오른쪽 : 반면 이를 불러오는 lightening의 코드는 언더바 없이 LRScheduler를 import하고 있다. 이러니 에러가 날 수밖에.

 

[해결법]

.conda/envs/{ENV_NAME}/lib/python3.8/site-packages/lightning_fabric/utilities/types.py로 들어가서 lightening의 LRScheduler import 코드에 언더바를 붙여서 저장해주면 된다.

 

저작자표시 (새창열림)

'Computer > 이게 왜 안되지?' 카테고리의 다른 글

[이왜안?] AttABseq code error  (0) 2025.09.12
[이왜안?] DiffAb 코드구현 에러발생 ValueError: invalid literal for int() with base 10: 'X'  (0) 2025.08.05
[이왜안] EOFError: EOF when reading a line  (0) 2025.06.23
[이왜안?] UnboundLocalError: cannot access local variable 'smallest_indices' where it is not associated with a value  (0) 2024.04.30
[이왜안?] PicklingError: Can't pickle <function <lambda> at 0x000002AC976E6CA0>: attribute lookup <lambda> on __main__ failed  (1) 2024.04.18
댓글