首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >使用val_loss而不是单独的绘图为每个时代获取对齐的WandB和train_loss图

使用val_loss而不是单独的绘图为每个时代获取对齐的WandB和train_loss图
EN

Stack Overflow用户
提问于 2022-03-11 00:46:51
回答 1查看 675关注 0票数 0

我有下面的代码记录火车和val损失在每一个时代使用WandB API。我不知道为什么我没有得到瓦尔损失和在同一时代的火车损失。知道怎么解决吗?

代码语言:javascript
复制
wandb.log({"train loss": train_epoch_loss,
           "val loss": val_epoch_loss,
           "epoch": epoch})

wandb.log({"train acc": train_epoch_acc,
           "val acc": val_epoch_acc,
           "epoch": epoch})

wandb.log({"best val acc": best_acc, "epoch": epoch})

正如你所看到的,val损失与时代和训练损失与时代是两个完全独立的实体,而我希望两者都在WandB中的一个情节中。

EN

回答 1

Stack Overflow用户

发布于 2022-03-14 11:00:53

我工作在重量和偏见,很高兴帮助:

2指标在相同图表上的

要在同一张图表上绘制2个指标,您可以单击图表中的铅笔图标来编辑它,然后将其他指标添加到y轴,如下所示。

更改默认的x轴

您还可以更改X轴,以绘制对应于“纪元”的图,而不是默认的wandb step。如果您希望默认情况下的这种行为,可以在开始训练之前调用wandb.define_metric一次,并将x轴设置为epoch。有关更多信息,请参见指标文档

测井步骤

需要注意的一件事是,当您记录您的验证度量时,您希望它们在与列车度量相同的步骤中被记录。在这种情况下,您可以这样做:

代码语言:javascript
复制
metrics = {}

for step, batch in my_data:
  ...
  
  train_loss = ....

  metrics['train_loss'] = train_loss

  if step % val_steps == 0:
    val_loss = ...
  
    metrics['val_loss'] = val_loss
  
    wandb.log(metrics)

  else:
    wandb.log(metrics)

或者,您可以在不增加wandb步骤的情况下使用论据wandb.log中存储度量。然后在需要时调用wandb.log()来增加步骤。

票数 3
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/71432453

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档