在 Hugging Face 使用 ML-Agents

ml-agents 是一個開源工具包，它使使用 Unity 製作的遊戲和模擬能夠作為訓練智慧代理的環境。

探索 Hub 中的 ML-Agents

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

Hub 上的所有模型都附帶了有用的功能

1. 一個自動生成的模型卡片，包含描述、訓練配置等資訊。
2. 有助於發現的元資料標籤。
3. Tensorboard 摘要檔案用於視覺化訓練指標。
4. 一個指向 Spaces 網路演示的連結，您可以在瀏覽器中視覺化您的代理進行遊戲。

安裝庫

要安裝 ml-agents 庫，您需要克隆倉庫

# Clone the repository
git clone https://github.com/Unity-Technologies/ml-agents

# Go inside the repository and install the package
cd ml-agents
pip3 install -e ./ml-agents-envs
pip3 install -e ./ml-agents

使用現有模型

您可以使用 mlagents-load-from-hf 從 Hub 下載模型。

mlagents-load-from-hf --repo-id="ThomasSimonini/MLAgents-Pyramids" --local-dir="./downloads"

您需要定義兩個引數

• --repo-id: 您要下載的 Hugging Face 倉庫的名稱。
• --local-dir: 下載模型的路徑。

視覺化代理遊戲

您可以直接在瀏覽器中輕鬆觀看任何模型進行遊戲

1. 轉到您的模型倉庫。
2. 在 Watch Your Agent Play 部分，單擊連結。
3. 在演示中，在步驟 1 中，選擇您的模型倉庫，即模型 ID。
4. 在步驟 2 中，選擇您要重播的模型。

分享您的模型

您可以使用 mlagents-push-to-hf 輕鬆上傳您的模型

mlagents-push-to-hf --run-id="First Training" --local-dir="results/First Training" --repo-id="ThomasSimonini/MLAgents-Pyramids" --commit-message="Pyramids"

您需要定義四個引數

• --run-id：訓練執行 ID 的名稱。
• --local-dir: 模型儲存的位置。
• --repo-id: 您要建立或更新的 Hugging Face 倉庫的名稱。格式為 <您的 huggingface 使用者名稱>/<倉庫名稱>。
• --commit-message.

額外資源

• ML-Agents 文件
• 官方 Unity ML-Agents Spaces 演示