目前,本人已经验证 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,就可以看到项目的训练情况:
查看硬件信息
扫描二维码关注公众号,回复:
14778701 查看本文章
创建自己的Wandb账户
wandb官网:https://wandb.ai/
wandb官方给的配置教程:MMDetection - Documentation(中文版) MMDetection - Documentation(英文版) 英文版有动态图片,比较直观。
首先你要去官网注册一个账号,然后去授权页面拿到私钥:Weights & Biases Authorize
然后在服务器安装wandb,并且用私钥登录
pip install wandb
wandb login
然后,再使用最开始的教程就可以了。
好了,大功告成,完结撒花!
欢迎关注、点赞、收藏、评论、分享给好友,一起学习有趣的新知识!!!