python - 複数の行を wandb にエクスポートする方法

Question

ライブラリの重みとバイアスを使用しています。私のモデルは曲線 (時系列) を出力します。この曲線がトレーニング中にどのように変化するかを確認したいと思います。そのため、エポックを選択できるスライダーが必要で、そのエポックの曲線が表示されます。これは、ヒストグラムで行われることと非常に似ている可能性があります (エポック全体のヒストグラムの画像が表示され、ホバーすると、そのエポックに対応するヒストグラムが表示されます)。を使用してこれまたは同様のことを行う方法はあり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)

助けていただければ幸いです！ありがとう！