很久沒有學了,期間打點滴打了一個多星期,太累了,再加上學了一下Python語法基礎,再終于開始重新學習pytorch
tensorboard 的使用
import torch
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter("logs")
for i in range(100):writer.add_scalar("y = 2x", 2*i,i)
# writer.add_scalar("image", torch.randn(1, 3, 224, 224))
writer.close()
在當前文件夾下的終端輸入?tensorboard --logdir=E:\pytorch_learn\logs?
writer.add_scalar("y = 2x", 2*i,i)
第一個參數是名字,第二個是y軸,第三個是x軸
?
writer.add_image("test", img_array, 2,dataformats="HWC")這個是顯示圖像 第一個參數是名字,第二個是圖像,第三個是訓練的步數,第四個是根據第二個參數的圖像格式來決定
?