Pytorch 視覺化之tensorboardX
阿新 • • 發佈:2018-12-12
使用tensorboardX需要安裝tensorflow和tensorboardX,安裝語句如下:
pip install tensorboardX
pip install tensorflow
其核心的語句為三句:
from tensorboardX import SummaryWriter
writer = SummaryWriter('./log/') # 設定儲存的log資料夾
writer.add_scalar('name', y, x) # 設定圖的名字,和y、x的值,注意這裡不是x,y,而是y,x
writer.close() # 關閉輸入
接下來我們來舉個栗子吧~
from tensorboardX import SummaryWriter if __name__ == '__main__': # x = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] writer = SummaryWriter('./log/') for i in range(1, 10): writer.add_scalar('test', i, i) writer.close()