wandb 使用

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查看