使用wandb进行模型训练可视化 |pytorch 神经网络 深度学习

wandb使用很简单，先去官网注册账号，获得一个key，复制

在命令行里

输入pip install wandb下载
wandb login 输入刚才获得的key登陆

在训练代码里面

导入

# Inside my model training code
import wandb
wandb.init(project="my-project")

声明超参数

wandb.config.dropout = 0.2
wandb.config.hidden_layer_size =