如何在 Tensorboard Pytorch 中繪制散點圖 plot？

Question

假設我想要一個在 TensorBoard 中繪制的通用散點圖 plot ，該散點圖繪制每個時期的第batch[:, 0], batch[:, 1] 。

如何在 TensorBoard 中做到這一點？

一個類似的老問題（2017 年 1 月）有一個解決方法，但我希望我們現在（2020 年 12 月）擁有真正解決方案的技術。

我的嘗試還不夠：

if self._current_epoch == 0:
    self.logger.experiment.add_scalars("epoch", {"batch": batch[:, 1]}, batch[:, 0])

給了我奇妙的錯誤

assert(scalar.squeeze().ndim == 0), '標量應該是 0D'

Answer 1

如果我正確理解您的問題，您可以使用add_images 、 add_figure將圖像或圖形添加到張量板（ 文檔）。

示例代碼：

from torch.utils.tensorboard import SummaryWriter
import numpy as np
import matplotlib.pyplot as plt

# create summary writer
writer = SummaryWriter('lightning_logs')

# write dummy image to tensorboard
img_batch = np.zeros((16, 3, 100, 100))
writer.add_images('my_image_batch', img_batch, 0)

# write dummy figure to tensorboard
plt.imshow(np.transpose(img_batch[0], [1, 2, 0]))
plt.title('example title')
writer.add_figure('my_figure_batch', plt.gcf(), 0)
writer.close()