1. 程式人生 > >Pytorch 視覺化之tensorboardX

Pytorch 視覺化之tensorboardX

使用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()

在這裡插入圖片描述