Timm 제공 모델 확인

Timm 이란

• Deep-learning 라이브러리로 Computer Vision 분야의 SOTA 모델들을 제공한다.
• optimizers, schedulers, data-loaders, augmentations 등 다양한 기능들을 제공한다.

Install

pip install timm

List Models with Pretrained Weight

pretrained_model_list = timm.list_models(pretrained=True)

print(pretrained_model_list)

Search for model achritectures by Wildcard

resnet_model_list = timm.list_models('*resnet*')

Create ResNet34 model

model = timm.create_model('resnet34')

Reference

• https://timm.fast.ai/