Hugging Face에서 fastai 사용하기

Hugging Face에서 fastai 사용하기 (Using fastai at Hugging Face)

fastai는 PyTorch와 Python을 활용해 텍스트, 비전, 테이블 데이터에 대해 최신 수준의 결과를 내는 빠르고 정확한 신경망을 학습할 수 있는 고수준 컴포넌트를 제공하는 오픈소스 딥러닝 라이브러리예요.

출처: 문서

본문

fastai는 PyTorch와 Python을 활용해 텍스트, 비전, 테이블 데이터에 대해 최신 수준의 결과를 내는 빠르고 정확한 신경망을 학습할 수 있는 고수준 컴포넌트를 제공하는 오픈소스 딥러닝 라이브러리입니다.

Hub에서 fastai 살펴보기 (Exploring fastai in the Hub)

models 페이지 왼쪽에서 필터링하면 fastai 모델을 찾을 수 있어요.

Hub의 모든 모델에는 다음 기능이 함께 제공됩니다:

검색 가능성에 도움이 되는 간단한 설명과 메타데이터 태그가 있는 자동 생성 모델 카드.
브라우저에서 모델을 직접 만져볼 수 있는 대화형 위젯 (Image Classification용).
추론 요청을 할 수 있는 Inference Providers 위젯 (Image Classification용).

기존 모델 사용하기 (Using existing models)

huggingface_hub 라이브러리는 Hub에서 모델을 다운로드하는 유틸리티 함수가 있는 가벼운 Python 클라이언트입니다.

pip install huggingface_hub["fastai"]

라이브러리를 설치했다면 from_pretrained_fastai 메서드만 사용하면 됩니다. 이 메서드는 모델을 불러올 뿐 아니라 모델이 저장된 당시의 fastai 버전도 검증하는데, 재현성에 중요해요.

from huggingface_hub import from_pretrained_fastai

learner = from_pretrained_fastai("espejelomar/identify-my-cat")

_,_,probs = learner.predict(img)
print(f"Probability it's a cat: {100*probs[1].item():.2f}%")

# Probability it's a cat: 100.00%

특정 모델을 불러오는 방법을 보려면 Use in fastai를 클릭하면 바로 사용할 수 있는 코드 스니펫이 주어져요!

push_to_hub_fastai 메서드로 fastai 모델을 공유할 수 있어요.

from huggingface_hub import push_to_hub_fastai

push_to_hub_fastai(learner=learn, repo_id="espejelomar/identify-my-cat")

추가 리소스 (Additional resources)

더 알아보기 (Learn more)

from_pretrained_fastai("repo_id")로 fastai 모델을 불러오고, push_to_hub_fastai(learner=learn, repo_id="...")로 내 모델을 Hub에 공유할 수 있어요. 모델 저장 당시의 fastai 버전 검증이 자동으로 이루어져 재현성이 보장됩니다. 자세한 내용은 fastai 코스와 공식 문서를 참고하세요.

Hugging Face에서 fastai 사용하기