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，具體可以參考：