什么是SwanLab?
SwanLab是一个深度学习实验管理与训练可视化工具,由西安电子科技大学创业团队打造,融合了Weights & Biases与Tensorboard的特点,可以记录整个实验的超参数、指标、训练环境、Python版本等,并可视化图表,帮助你分析训练的表现。
SwanLab也与绝大多数主流的训练框架进行了集成,包括PyTorch、HuggingFace Transformers、PyTorch Lightning、MMDetection、Ultralytics、fastai等等,可以开箱即用地与你的深度学习工作流轻松集成。
手机上查看实验
一次稍大的AI实验,往往需要在服务器上运行几个小时甚至几天的时间————所以OpenAI算法工程师JasonWei在自己的日程表上,常常把训练模型放在睡觉前启动,并在醒来后查看结果。
在训练过程中,你可能在健身、在通勤、在吃饭而无法看到当前实验的最新情况,这个时候如果能够在手机、平板上看到实验的最新进展,及时发现训练的问题,能够节省很多的时间。而且这种对新实验进展的好奇心,我相信每一位炼丹师都懂!
而恰好,SwanLab就是一个在云端记录AI实验的工具(而且服务器在国内,没有网络问题),能够支持你随时随地看实验 ———— 只需将SwanLab嵌入到你的训练代码中即可:
"""简单嵌入案例"""
import swanlab
# 启动SwanLab
swanlab.init(project="my-ml-train")
···
# 记录指标
swanlab.log({"loss": loss, "acc": acc})
接下来介绍两种在手机上看实验的方法。
(推荐) 方法一:使用PWA安装SwanLab
如果你的手机是IOS系统,那么打开Safari浏览器 -> 访问SwanLab网站 -> 点击底部的分享按钮 -> 添加到主屏幕 -> 添加,然后在你的主屏幕上会出现一个SwanLab App,之后每次打开它即可。
如果你的手机是安卓系统,那么在自带的浏览器上一般都会有将网页添加到桌面的功能,执行即可。
方法二:微信公众号访问
通过微信来访问SwanLab也是种非常方便的方法,可以通过公众号来实现这一点。