[PyTorch] torch._six Issue

안넝

이번에 YOLOv4 pretrain하는 코드를 돌려보면서 디버깅하다가 찾은 내용을 공유하고자 한다.

PyTorch 2.0 이상에서는 이제 3.8+ 버전의 Python만 지원하기때문에 더이상 torch._six 라이브러리를 지원하지 않는다 !

error

 

관련하여 아래와 같이 pytorch 깃허브에도 내용이 있다.

PyTorch 공식 Github

 

친절하게 1.13.1과 비교해서 2.0.0 이상의 버전에서는 어떤 식으로 대체해야할지도 알려준다 !

대체 사용법

 

관련하여 torch._six.string_classes 코드를 나같은 경우 torch.StringType로 교체하여 해결했다.

 

끝 !