wandb 使用

最新推荐文章于 2025-11-18 00:38:21 发布

原创最新推荐文章于 2025-11-18 00:38:21 发布 · 6.1k 阅读

16 ·

CC 4.0 BY-SA版权

文章标签：

#人工智能 #机器学习 #python

深度学习专栏收录该内容

8 篇文章

订阅专栏

本文介绍了如何注册wandb账户，获取API密钥，并在Python环境中安装wandb库。通过设置wandb配置，初始化项目，跟踪超参数和运行元数据，可以对训练过程中的loss和指标进行日志记录，便于在wandb平台上可视化分析。

GPT-oss:20b

图文对话

Gpt-oss

GPT OSS 是OpenAI 推出的重量级开放模型，面向强推理、智能体任务以及多样化开发场景

1. 注册

wandb官网 https://wandb.ai/

2. 登录后在User setting页面获取 API key 并复制

在这里插入图片描述

3. 安装wandb

pip install wandb

4. 在代码中使用wandb

os.environ['WANDB_API_KEY'] = '6cabcsg...(上面获取的API key)'
import wandb
wandb.login()

5. 初始化

wandb.init(
        # Set the project where this run will be logged
        project="project name",
        # Track hyperparameters and run metadata
        config={
            "learning_rate": 0.0001,
            'hidden_size': 128, 
            ...
        },
        # 每次运行的名字
        name = dataset_name
        )

6. 日志

想要观测的数据

wandb.log({'train_loss': train_loss, 'val_loss': val_loss, 'val_mcc': val_mcc})

7. 回到wandb网页查看

选择刚才创建的项目
在这里插入图片描述
选择指定的name查看

在这里插入图片描述

您可能感兴趣的与本文相关的镜像

GPT-oss:20b

图文对话

Gpt-oss

GPT OSS 是OpenAI 推出的重量级开放模型，面向强推理、智能体任务以及多样化开发场景