0


wandb 安装与使用

wandb(Weights & Biases)是一个类似于tensorboard的在线模型训练可视化工具。

1)注册和安装wandb

注册wandb
到其官网 https://wandb.ai/home 注册
安装wandb
执行:

pip install wandb

在这里插入图片描述

wandb login

这个输入的时候,貌似不能paste,只能一个个地手动输入
注意API 输入的是,登录wandb后,个人界面里有一个API keys
在这里插入图片描述

在这里插入图片描述

2)安装wandb完之后

可以在C盘user目录中找到.netrc文件,里面有你的登录密码
在这里插入图片描述

3) 在代码中修改项目名称

在utils/wandb_logging/wandb_utils.py中,把project名字改成你的项目名

 self.wandb_run = wandb.init(config=opt,
                                    resume="allow",
                                    project='YOLOv5-Ball'if opt.project =='runs/train'elsePath(opt.project).stem,
                                    name=name,
                                    job_type=job_type,
                                    id=run_id)ifnot wandb.run else wandb.run
标签: 人工智能 pytorch

本文转载自: https://blog.csdn.net/Captain_Aaron/article/details/122765088
版权归原作者 Aaron_Yang. 所有, 如有侵权,请联系我们删除。

“wandb 安装与使用”的评论:

还没有评论