SwanLab

收录理由

用PyTorch或LLaMA Factory训练模型的团队，常常要花不少精力去复盘一次训练到底发生了什么。SwanLab用一个轻量的Python SDK解决了这个麻烦，把它加进训练脚本，就能把指标、超参数、硬件利用率以及图片、视频这类媒体内容都记录下来，呈现在一个可检索的看板上，整个团队都能浏览。它集成了50多个框架，通常不需要改变你原有的训练方式，只需加上几行代码，就能把本来已经产生的数据展示出来。既可以用云端托管服务，也可以自托管部署，小团队可以马上开始用，注重安全的企业则能把数据完全留在自己的服务器上。如果TensorBoard那种静态图表已经不够用，需要实验对比、分组、基线视图和现代风格的网页界面，SwanLab是一个很实在的升级选择。