Как экспортировать несколько строк в 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)

Буду признателен за любую помощь! Спасибо!

Answer 1

Ты можешь использовать с 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 для запуска, вы увидите, что сюжет визуализирован как сюжетный сюжет. Щелкните шестеренку в верхнем левом углу, чтобы увидеть ползунок, который позволяет перемещаться по этапам тренировки и просматривать график на каждом этапе.