안녕하세요 교수님

Question

교수님의 강의를 듣고 pytorch로 넘어와서 작업을 하다 tensorflow로 다시 작업을 해야할 상황이 생겼는데,,,,

pytorch 같은 경우 timm이라는 라이브러리를 통해 vit, convnext등 다양한 모델을 이용할 수 있는데 tensorflow같은 경우 어떻게 해야 해당 모델들을 쓸 수 있는지 궁금해서 이렇게 질문을 드립니다 :)

Answer

안녕하십니까,

음, timm... tensorflow하는 입장에서는 부럽기도 합니다. 안타깝게도 timm같이 서드파티에서 더 나은 성능을 가진 pretrained 모델을 찾기는 어렵습니다. tensorflow의 내장된 pretrained 모델을 사용할 수 밖에 없습니다.

다만 Transformer 기반의 VIT 같은 경우는 Hugging Face에서 생성된 모듈을 사용하시면 될 것 같습니다.

https://huggingface.co/docs/transformers/model_doc/vit 중간쯤에 TF 기반 VIT 사용법이 있습니다.

감사합니다.