timm 라이브러리1 [논문 구현] ViT ImageNet 평가 방법 ViT는 트랜스포머 중에서 그나마 간단한 형태이다. 실제로 구현하는게 그리 어렵지는 않다. 하지만.......... 논문에서 '대용량 pre-training'이 안된 ViT는 퍼포먼스가 상당히 떨어진다고 나온다. 다시 말해서 시간도 오래 걸리고 귀찮다................................. 다행히 나만 귀찮은게 아니라 천조국 형들도 귀찮았나 보다. 누군가 이미 라이브러리로 만들어놨다. https://timm.fast.ai/ 이걸로 모델을 만들어보자. 일단 설치 pip install timm 그리고 OOP로 작성된 코드는 다음과 같다. (device는 맥북이면 mps, 윈도우면 cuda) import timm import torch import torchvision import torc.. 2023. 3. 28. 이전 1 다음
