返回首页
详细介绍
导语:SwanLab是情感机器(北京)科技有限公司推出的模型训练研发工具,定位为AGI时代的先进AI实验平台。它面向AI训练过程,提供训练可视化、自动日志记录、超参数记录、实验对比与多人协同等专业能力。开发者只需少量代码即可接入,并支持三十多种主流训练框架,让训练过程变得可被记录与复盘。
核心能力
- 训练可视化以折线图、媒体图等多种图表实时呈现训练进展,辅助分析决策。实时看板让异常更早被发现。
- 自动日志记录会收集终端日志,并自动记录硬件信息、Python与Git等编程环境。环境信息对追溯很有价值。
- 超参数记录可归纳每次实验的配置并支持搜索,方便横向比较不同实验差异。参数一目了然便于复盘。
- 实验对比通过集中式表格呈现多组结果,帮助研究者启发灵感、逼近更优模型。对比中更容易找到方向。
- 多人协同支持不同训练师在同一项目跑实验,组织版可沉淀团队的训练经验。协作打破沟通壁垒。
适合谁
- 高校与科研院所的研究者需要系统记录实验,借助可视化找到迭代方向。图表比日志更直观。
- 企业算法团队希望打通沟通壁垒,用实验链接同步进展、提升协作效率。远程协作也能顺畅。
- 使用PyTorch、Transformers等框架的开发者,可用六行代码快速开始跟踪。接入成本低到几乎无感。
- 经常远程或移动办公的人,能用多端适配在手机上随时查看训练进展。通勤时也能掌握状态。
- 不便联网的环境可启用离线使用方式,满足本地化训练的记录需求。离线场景同样可追踪。
使用建议
- 开始训练前用 swanlab.init 创建项目并写入超参数,后续指标用 swanlab.log 记录。初始化一步不能漏。
- 多实验并行时用集中表格做对比,先筛选方向再投入更多算力。避免算力被低效实验浪费。
- 团队项目建议启用组织版,把训练经验沉淀为可复用的协作资产。经验沉淀比单打独斗高效。
- 关注硬件信息自动记录,出现训练异常时可快速定位显存或环境层面的问题。定位快才能止损快。
- 夜间使用开启黑夜模式,长时间盯看训练看板时更护眼。暗色界面减轻视觉疲劳。


