EfficientNet

[jinmang2]

EfficientNet

• 분류 모델로 EfficientNet 사용도 고려해보면 어떨까요?
• 분류에서 TOP은 아니지만 킹성비가 좋다고만 추천받았어요!
• https://github.com/rwightman/pytorch-image-models/blob/master/timm/models/efficientnet.py

[lsh3163]

말씀하신 문서를 보니 허깅페이스처럼 잘 정리가 되어있네요! (피어세션 때 잘 알아보지도 않고 답변했네요;;)

import torch
import timm
m = timm.create_model('resnet50', pretrained=True, num_classes=0)
o = m(torch.randn(2, 3, 224, 224))
print(f'Pooled shape: {o.shape}')

timm의 모델 객체는 nn.Module 을 따르고 있기 때문에 모델의 사이즈, 손실 함수 등만 잘 고려하면 저희가 잘 활용할 수 있을 것 같습니다.

모델 종류는 다음과 같은데 GPU사양에 따라 파라미터가 적은 순으로 사용하면 될 것 같습니다!

[jinmang2]

앙상블 모델 후보가 늘어가는군요 ㅎㅎㅎ huggingface에 beit, vit 류도 많이 있던데 다음주에 저희가 실험해볼 모델이 많은 것 같아서 기대가 큽니다!!