linux下使用tensorboadX可視化pytorch訓練結果

1、依賴

首先你需要安裝torch、tensorflow、tensorboardX

2、基本代碼

from tensorboardX import SummaryWriter
writer = SummaryWriter(logdir) # logdir是你自己的保存記錄的目錄
for i in range(epoch):
	loss = model(xxx)
	writer.add_scalar('train/loss', loss.item(), i)
writer.close()

3、使用tensorboadx

如果你有多個Python，需要指定你當前使用的python的tensorboard，比如：

python xxx/site-packages/tensorboard/main.py --logdir=/data02/gob/project/document_event_extraction/tensorboard --port=6006

我們在自己的python環境下找到site-packages/tensorboard/main.py所在的路徑，然后輸入python xxx/site-packages/tensorboard/main.py --logdir=xxx --port=6006指令即可。

4、可視化結果

我們在linux下難以輸入localhost:6006來在瀏覽器查看結果，因此，可以在window下建立一個映射，使得我們可以在window下查看linux下tensorboard的結果，這里我們需要的工具是xshell，具體可以參考：
https://blog.csdn.net/weixin_42059276/article/details/106668679
如果嘗試了還是不成功的話，就新建一個6006到6006的映射，然后在瀏覽器輸入：127.0.0.1:6006