Hugging Face 发布了 Trackio,这是一个轻量级、开源且免费的实验跟踪 Python 库。Trackio 提供本地仪表盘,并与 Hugging Face Spaces 无缝集成,方便用户分享和协作训练进度。该库设计为 wandb 的替代品,用户可以使用熟悉的语法快速上手。
Trackio 允许用户跟踪训练过程中的指标、参数和超参数,并通过可视化仪表盘进行分析。它支持从 nvidia-smi 命令获取 GPU 能耗数据,便于量化模型训练的环境影响。数据可轻松提取,用于自定义分析或集成到研究流程中。
用户可以通过 pip 安装 Trackio,例如使用命令 pip install trackio。在代码中,可以导入 trackio 作为 wandb 使用,兼容 wandb.init、wandb.log 和 wandb.finish 等 API。仪表盘可同步到 Hugging Face Spaces,通过 URL 分享,支持私有或公开设置。