wandb可视化工具部署本地化

author:张一极

[email protected]


1.关于wandb

引言:wandb是一个可私有化部署的实验可视化工具,具有实验过程记录,超参记录,以及其他重要参数可视化功能,可以用于可视化模型训练过程中ap变化情况,召回规律,精确率变化情况等:

image-20220330223324154

以及可视化一些保存下来的中间图像(比如mosaics结果等):

image-20220330223433549

2.部署过程

私有化部署支持docker一键部署,可配置参数

1.docker run --rm -d -v wandb:/vol -p 8080:8080 --name wandb-local wandb/local

docker跑起来以后,配置好端口

2.配置好host路径,为了配置下一步,wandb local --port 8960 -e HOST=http://xxx:port

3.配置好许可证,复制进system setting

4.完成许可授权以后,即可进入

5.在你的机器上,wandb login 输入你的令牌(在主页复制)

image-20220330222953898

6.会有提示:

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

7.在你的代码中按照doc内容执行wandb数据传输即可,详情见:https://colab.research.google.com/github/wandb/examples/blob/master/colabs/intro/Intro_to_Weights_%26_Biases.ipynb

如果无法登陆,需要在新机器上需要手动设置:登陆的同时配置一下host,wandb login --host=http://your_wandb_ip:port


PS:退出登录以后,重置登录密码:

以上