pytorch 中用tensorboard 可視化

Step1:

  假設已經安裝了tensorflow

• 安裝tensorboard: 

  pip install tensorboard

• 安裝tensorboardX：可通過

  pip install tensorboardX

  指令直接安裝。

Step2:使用舉例

import numpy as np 
from tensorboardX import SummaryWriter
writer=SummaryWriter('yourfolder') 
for epoch in range(100): 
         writer.add_scalar('scalar/test',np.random.rand(),epoch)         

         writer.add_scalars('scalar/scalars_test',{'xsinx':epoch*np.sin(epoch), \
'xcosx':epoch*np.cos(epoch)},epoch) 
writer.close() 
           

Stp3:檢視方法

       首先終端： tensordboard --logdir=yourfolder

        打開浏覽器： 輸入：http://localhost:6006

參考：https://www.jianshu.com/p/46eb3004beca