timm library로 인한 RuntimeError 발생

RuntimeError 발생

 

제대로 실행되는 code를 가지고 새롭게 test진행을 하려고 anaconda의 새로운 가상 환경을 만들었다.

 

code를 돌리기 위해 필요한 library들을 설치하고 test를 진행했는데 갑자기 아래와 같이 error가 발생하는 것이다.

 

error message: 

RuntimeError: Given normalized_shape=[1024], expected input with shape [*, 1024], but got input of size[1]

 

 

해결 방법

위의 error을 해결하기 위해 anaconda의 가상 환경도 다시 만들어 보고 pytorch verion과 torchvision version도 여러 번 바꿔가기도 해봤는데 원인은 전혀 다른 library에 대한 문제였다.

 

timm library의 version 문제였던 것이다.

 

사용한 model에 대한 github 주소에 가서 README.md를 보는데 설치 됐었던 timm의 version이 README.md에 적혀있는 version과 다른 version이였다.

 

그래서 혹시 몰라 맞춰서 설치를 했더니 해당 오류가 해결됐다.

 

 

timm library란

timm library는 hugging face에서 만든 library로 다음과 같은 computer vision에 대한 SOTA들이 포함되어 있다.

• models / layers / utilities / optimizers / schedulers / data-loaders / augmentations / training-evaluation scripts.

자세한 설명은 밑의 링크에 들어가보면 설명되어 있다.

 

 

https://huggingface.co/docs/timm/index

timm