【wandb与MMcv】Using wandb in MMcv，查看模型训练

        目前，本人已经验证  MMCV中的MMdetection、MMrotate支持使用wandb查看模型训练情况，可以直接在【configs/_base_/default_runtime.py】下添加如下代码：

本教程需要有自己wandb账户， 没有wandb账户划到文章最后看如何创建        

       dict(
            type='WandbLoggerHook',
            init_kwargs=dict(
                project='mmrotate',
                name='cfa-resnet101'
            )
        )

添加好的效果如下:

 然后，再运行自己的代码，训练模型就可以看到下面的显示了：

最后两个链接，是你wandb账户的运行页面。

简单介绍wandb使用

打开自己的wandb就可以看到已经创建好了你要运行的项目：

 

点击自己的Runs下的Name，就可以看到项目的训练情况：

 查看硬件信息

 

---

 创建自己的Wandb账户

wandb官网：https://wandb.ai/

wandb官方给的配置教程：MMDetection - Documentation（中文版） MMDetection - Documentation（英文版） 英文版有动态图片，比较直观。

        首先你要去官网注册一个账号，然后去授权页面拿到私钥：Weights & Biases Authorize

        然后在服务器安装wandb，并且用私钥登录

pip install wandb
wandb login

然后，再使用最开始的教程就可以了。

好了，大功告成，完结撒花！

欢迎关注、点赞、收藏、评论、分享给好友，一起学习有趣的新知识！！！