如何将多行导出到 wandb

Question

我正在使用库权重和偏差。我的模型输出一条曲线（时间序列）。我想看看这条曲线在整个训练过程中是如何变化的。所以，我需要某种滑块，我可以在其中显示 select 个纪元，它会显示那个纪元的曲线。它可能与它对直方图所做的非常相似（它显示跨时期的直方图图像，当您将鼠标悬停时，它会显示与该时期对应的直方图）。有没有办法使用 wandb 来做到这一点或类似的事情？

目前我的代码是这样的：

for epoch in range(epochs):
   output = model(input)
   #output is shape (37,40) (lenght 40 and I have 37 samples)
   #it's enough to plot the first sample
   xs = torch.arange(40).unsqueeze(dim=1)
   ys = output[0,:].unsqueeze(dim=1)
   wandb.log({"line": wandb.plot.line_series(xs=xs, ys=ys,title="Out")}, step=epoch)

如有任何帮助，我将不胜感激！谢谢！

Answer 1

您可以将 wandb.log() 与 matplotlib 一起使用。使用 matplotlib 创建绘图：

import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(0, 50)
for i in range(1, 4):
    fig, ax = plt.subplots()
    y = x ** i
    ax.plot(x, y)
    wandb.log({'chart': ax})

然后，当您查看运行的 wandb 仪表板时，您会看到呈现为 plotly plot 的绘图。单击左上角的齿轮以查看滑块，您可以在其中滑动训练步骤并查看每个步骤的绘图。

如何将多行导出到 wandb

How to get multiple lines exported to wandb

python

pytorch

wandb