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.initwandb.logwandb.finish 相容,易於直接替換。
  • LLM 友善: 包含用於程式化存取和直接 SQL 查詢實驗資料的 CLI。
  • 本地優先設計: 無需帳戶即可開始記錄;資料預設保留在本地。
  • 高吞吐量: 非阻塞架構允許每秒進行數千次日誌呼叫。
  • 可擴充: 使用 Python 和 Svelte 5 構建,易於 fork 並自訂後端和前端儀表板。

相關

  • Dispatch
  • Dispatch
  • 專案
  • 專案
  • Dispatch