PyTorch TensorBoard 寫入頻率

Question

我正在嘗試使用火炬 (torch.utils.tensorboard) 將我的訓練和驗證損失寫入張量板，它看起來最多只能寫入 1000 個數據點，無論實際迭代次數是多少。 例如，運行以下代碼，

writer1 = SummaryWriter('runs/1')
writer2 = SummaryWriter('runs/2')

for i in range(2000):
    writer1.add_scalar('tag', 1, i)

for i in range(20000):
    writer2.add_scalar('tag', 1, i)

在檢查和下載 csv 時，兩者都精確地產生 1000 個點，即使在 tensorboard 儀表板上，第一個點從第 5 步和第 18 步開始並遞增，以便總步數為 1000，而不是 2,000 和 20,000。

我不知道這是 tensorboard 的默認行為還是 PyTorch 的決定，但無論哪種方式，有沒有辦法編寫每一步？

Answer 1

其實我在這里找到了答案。 因此，SummaryWriter 在每個時期都進行保存，但要加載所有內容，必須以--samples_per_plugin scalars=0標志啟動--samples_per_plugin scalars=0 。 0 告訴 tensorboard 加載所有點，而 100 表示總共 100 個點，例如

總結一下，我用命令tensorboard --logdir=logs --samples_per_plugin scalars=0啟動了tensorboard