前言

本文介绍wadb基本使用方法


一、安装wandb及登陆

安装wandb

(myenv)pip install wandb

wandb官网创建账号,我的是直接链接到github
登陆wandb:

(myenv)wandb login

出现以下提示:
在这里插入图片描述
可以按照提示的网址登陆https://wandb.ai/authorize获取API key或者如下图所示步骤获取
在这里插入图片描述
在这里插入图片描述

输入API key后得到:
在这里插入图片描述

说明一下我在docker 中输入API key之后得到的是如下提示:

wandb: Appending key for api.wandb.ai to your netrc file: /root/.netrc

这时候只要按提示创建/root/.netrc文件并将API key存入其中即可,注意.netrc文件中除了API key之外不要有任何空格、换行符或其他字符

二、编写python脚本

import wandb
run = wandb.init(project="your_project_name")# 初始化
run.log({"data": data, "loss": loss})  #记录数据
run.watch(model, log="all") #查看结果,这行代码也可以不要,直接到wandb账号中的project里面查看
run.finish() #结束
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐