timm 모델 목록 및 구조보기

Leejaegun·2024년 11월 28일
0

Python & etc

목록 보기
7/27

timm은 라이브러리

1. 모델 무엇 있는지 보기

import timm
for model in timm.list_models():
	print(model)

2. 모델 아키텍쳐 보기

import timm

# 모델 불러오기 (예: resnet50)
model_name = "resnet50"
model = timm.create_model(model_name, pretrained=True)
print(model)
profile
Lee_AA

0개의 댓글