三行代码，教你做着这种训练损失图像监控！研一，研二，研三，必看！wandb，一个监控简易训练库，论文作图必看，论文写作

前言

提示：内容非常简单：

---

一、wandb是什么？

示例：这里不废话。

二、使用步骤

1.引入库

先安装下载一个库 wandb：

pip install wandb

2.登录注册一个wandb账号

在运行wandb login之前最好先去wandb的官网注册一个账号，这样在输入API key时可以方便不少。点这里注册

注册登录之后，会得到一个key：

然后cmd输入以下命令，代码如下（示例）：

wandb login

注意：将刚才复制的key粘贴到这里，！！！！粘贴之后不显示密码是正常的！！！！粘贴之后不显示密码是正常的！！！！粘贴之后不显示密码是正常的！！！！粘贴之后不显示密码是正常的！！！！粘贴之后不显示密码是正常的！！！！粘贴之后不显示密码是正常的！！！！粘贴之后不显示密码是正常的！！！！粘贴之后不显示密码是正常的！！！！粘贴之后不显示密码是正常的！！！！粘贴之后不显示密码是正常的！！！！粘贴之后不显示密码是正常的！！！！

---

正常使用

这里只需要三行代码，即可完成监控效果，
第一行

wandb.init(project='你的项目名，自己起一个',
           name='你的项目下的子项目',
           config=config,)

上面的config，是一个字典用于存储训练时的参数
第二行

            wandb.log({
                "loss_s":listloss[0].item(),
                "loss_i":listloss[1].item(),
                "loss_d":listloss[2].item(),
                "loss_mm":listloss[3].item(),
                "loss_contrastive":loss_contrastive.item(),
                'loss':loss.item(),
                'epoch':epoch
            })

构建你的log，这就是你想要监控的参数的数值，注意config.log(这里面也要一个字典)
第三行
好像没有第三行了，这样就可以启动了。运行之后，会在命令行输出你得wandb监控地址
我们选择第二个

打开浏览器之后，就是美美的图片了。

总结

欢迎大家关注我的***宫中皓***，往期文章可以看到，或者私信我。谢谢大家。