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

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

创新互联公司自2013年起，是专业互联网技术服务公司，拥有项目网站设计、成都网站设计网站策划，项目实施与项目整合能力。我们以让每一个梦想脱颖而出为使命，1280元井陉做网站,已为上家服务,为井陉各地企业和个人服务,联系电话:18980820575在命令行里

输入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 = 128

记录指标

def my_train_loop():
    for epoch in range(10):
        loss = 0 # change as appropriate :)
        wandb.log({'epoch': epoch, 'loss': loss})

保存文件

# by default, this will save to a new subfolder for files associated
# with your run, created in wandb.run.dir (which is ./wandb by default)
wandb.save("mymodel.h5")

# you can pass the full path to the Keras model API
model.save(os.path.join(wandb.run.dir, "mymodel.h5"))

参考

快速上手 https://docs.wandb.ai/v/zh-hans/quickstart
示例: 官方示例代码

你是否还在寻找稳定的海外服务器提供商？创新互联www.cdcxhl.cn海外机房具备T级流量清洗系统配攻击溯源，准确流量调度确保服务器高可用性，企业级服务器适合批量采购，新人活动首月15元起，快前往官网查看详情吧

当前文章：使用wandb进行模型训练可视化|pytorch神经网络深度学习-创新互联
网页地址：http://cdkjz.cn/article/hesih.html