인프런 커뮤니티 질문&답변
작성한 질문수
안녕하세요 교수님
작성
·
443
0
교수님의 강의를 듣고 pytorch로 넘어와서 작업을 하다 tensorflow로 다시 작업을 해야할 상황이 생겼는데,,,,
pytorch 같은 경우 timm이라는 라이브러리를 통해 vit, convnext등 다양한 모델을 이용할 수 있는데 tensorflow같은 경우 어떻게 해야 해당 모델들을 쓸 수 있는지 궁금해서 이렇게 질문을 드립니다 :)
퀴즈
What does it mean the closer the ROC AUC score, which is used as the main indicator for evaluating model performance in a plant pathology competition, is to 1?
It means the model's predictive performance is close to random.
The model's predictive performance is excellent.
Clear evidence of model overfitting.
This means that the Learning Rate is set too high.
답변 1
0
안녕하십니까,
음, timm... tensorflow하는 입장에서는 부럽기도 합니다. 안타깝게도 timm같이 서드파티에서 더 나은 성능을 가진 pretrained 모델을 찾기는 어렵습니다. tensorflow의 내장된 pretrained 모델을 사용할 수 밖에 없습니다.
다만 Transformer 기반의 VIT 같은 경우는 Hugging Face에서 생성된 모듈을 사용하시면 될 것 같습니다.
https://huggingface.co/docs/transformers/model_doc/vit 중간쯤에 TF 기반 VIT 사용법이 있습니다.
감사합니다.
이것저것 검색하는 도중 tfimm(https://github.com/martinsbruveris/tensorflow-image-models) 라이브러리 발견했는데,,,꽤 유용한거같습니다!