第一次接触Wandb
01.Wandb的介绍
wandb(Weights & Biases)是一个流行的工具,用于机器学习项目的实验跟踪、可视化和管理。它可以帮助研究人员、数据科学家和开发者更有效地监控他们的机器学习模型的训练过程,优化模型性能,以及分享和复现实验结果。下面是wandb的一些主要功能:
实验跟踪
-
自动日志记录:自动记录训练过程中的超参数、训练损失、验证损失和任何其他自定义指标。
-
实时更新:在训练过程中实时查看图表和指标,使得可以即时发现问题并调整训练策略。
可视化
- 生成图表:创建关于模型性能的交互式图表,例如损失曲线和精度曲线。
- 数据和模型可视化:可视化模型的预测结果、权重分布或特征图等。
模型管理
- 保存和共享:保存模型的所有相关文件,包括权重文件、配置文件和环境设置,以便轻松地与他人共享或重新训练。
- 版本控制:对模型版本进行跟踪,比较不同模型的性能。
协作和报告
- 团队协作:在团队或社区中共享项目,使团队成员可以查看实验结果和进度。
- 生成报告:创建可分享的可视化报告,以交流实验结果和见解。
02.Wandb 的使用
首先进入官网进行账号注册。如果有Google或者GitHub账号可以直接登陆。
网址:Weights&Biases官网
注册成功之后就会看到如下界面:
在pycharm终端中,先运行pip install wandb,再输出wandb login
Debug:无法粘贴API key
其实不是无法粘贴,而是不可见。
我们要做的就是复制一次,再粘贴一次
就可以了!
登录成功的界面如下所示: