在 Hugging Face 使用 fastai

fastai 是一個開源的深度學習庫，它利用 PyTorch 和 Python 提供高階元件，用於在文字、視覺和表格資料上訓練快速準確的神經網路，並生成最先進的輸出。

在 Hub 中探索 fastai

您可以透過在模型頁面左側進行篩選來查詢 fastai 模型。

中心上的所有模型都具有以下功能：

1. 自動生成的模型卡片，其中包含簡要說明和有助於發現的元資料標籤。
2. 一個互動式小部件，您可以直接在瀏覽器中（用於影像分類）使用該模型進行演示。
3. 一個推理 API，允許進行推理請求（用於影像分類）。

使用現有模型

huggingface_hub 庫是一個輕量級的 Python 客戶端，具有從 Hub 下載模型的實用函式。

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")

其他資源

• fastai 課程。
• fastai 網站。
• 與 Hub 整合文件。
• 與 Hub 整合公告。