gradio-app/trackio
A lightweight, local-first, and free experiment tracking library from Hugging Face 🤗
解決的問題
Trackio 是一個輕量級、免費的實驗追蹤函式庫,旨在取代笨重且依賴雲端的工具。它透過提供一個無需建立帳戶的本地優先日誌系統,以及為平行實驗提供高吞吐量的資料儲存,專門滿足了人類研究人員和 AI 智能體的需求。
運作方式
Trackio 將指標、配置和媒體記錄到本地 SQLite 資料庫中,為了效率可以將其「凍結」為 Parquet 檔案。它使用非阻塞的背景執行緒每 0.5 秒批次寫入一次日誌,確保訓練迴圈不會被 I/O 拖慢。在視覺化方面,它提供了一個受 Gradio 啟發的儀表板,可以在本地執行、在私人基礎設施上自行託管,或者免費部署到 Hugging Face Spaces 以進行協作,並透過 IFrames 嵌入網站。
適用對象
- ML 研究人員: 尋求免費、本地優先的 Weights & Biases 替代方案,且需要熟悉 API 的人。
- AI 智能體開發者: 正在構建自主 ML 實驗,需要可程式化 CLI 和 SQL 介面來查詢實驗資料的開發者。
- 協作團隊: 希望透過 Hugging Face Spaces 分享即時儀表板,而無需管理複雜基礎設施的使用者。
亮點
- W&B 相容: API 與
wandb.init、wandb.log和wandb.finish相容,易於直接替換。 - LLM 友善: 包含用於程式化存取和直接 SQL 查詢實驗資料的 CLI。
- 本地優先設計: 無需帳戶即可開始記錄;資料預設保留在本地。
- 高吞吐量: 非阻塞架構允許每秒進行數千次日誌呼叫。
- 可擴充: 使用 Python 和 Svelte 5 構建,易於 fork 並自訂後端和前端儀表板。
相關
- Dispatch
- Dispatch
- 專案
- 專案
- Dispatch