站点图标 AI技术聚合

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' else Path(opt.project).stem,
                                    name=name,
                                    job_type=job_type,
                                    id=run_id) if not wandb.run else wandb.run

文章出处登录后可见!

已经登录?立即刷新
退出移动版