PyTorch训练可视化工具：Tensorboard、SwanLab、Wandb、MLFlow特点分析

原创

已于 2025-03-27 01:33:29 修改 · 2k 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #人工智能 #python #深度学习 #机器学习

于 2025-01-09 11:58:07 首次发布

tensorboard

swanlab

wandb.ai

mlflow

1. 🇺🇸Tensorboard: 老牌工具，尚能饭否

Tensorboard，包括用于 Pytorch 的 TensorboardX，Tensorboard 本身被设计成插件化的方式，好处是轻量、轻耦合，可以按需要很快的自定义一个新的 tab，这种方式的缺点是缺少一个用户整体的视图，用户很难在宏观视角上对整个训练流程有一个比较清楚的认识。

特点:

出的比较早，格式通用，但功能相对单薄，用户体验已跟不上新的工具
虽然是跟 Tensorflow 出的，但并不仅限于 Tensorflow，Pytorch也能用（通过TensorboardX）
实验全流程记录
报表和跟其他系统的兼容性比较好，因为出的早?..
适合单人使用，没有团队功能
开源项目里带的比较多

‍

2. 🇨🇳SwanLab: 国产新星，简洁易用

SwanLab是2024年推出的深度学习可视化工具，来自一支年轻的中国团队，支持云端+离线两种使用方式。SwanLab的UI设计应该是所有这类工具里面最现代的、交互最阳间的（类似Vercel的Next设计风格），Python API的设计基本上对标Wandb做的，综合实力不错。不过因为比较新的关系，一些高级功能还有待完善。</