torch._six error 해결

pytorch 여러버전 깔아봤는데 똑같이 에러 발생

 

~/lib/python3.10/site-packages/timm/models/layers

들어가서

 

helpers.py 안에서

 

#from torch._six import container_abcs
import collections.abc as container_abcs

수정하면 동작 OK