ViT Pretrained Model 불러오기

timm 설치 후 간단하게 pretrained model을 불러올 수 있다.

ViT pretrained model을 가져와 classification을 진행해보았다.

 

import timm

model = timm.create_model('vit_base_patch16_384', pretrained=True) 
model.eval()