17
13

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 5 years have passed since last update.

tensorboardXの簡単な使い方まとめ

Last updated at Posted at 2018-11-23

#TensorboardXとは
pytorchでtensorboardを使いたい人向けのライブラリ

##簡単なイメージ
tensorboardを使ったことがない人向けに説明すると

モデルの学習中に出てくるlossなどをAPIを使ってファイルに出力

pathは出力されたファイルのディレクトリ
$ tensorboard --logdir="path"

コマンドでtensorboard起動

ブラウザでhttp://localhost:6006
にアクセスすると(localhostなどはリモートPCの場合適宜変えてください)

こんな画面が出てくる

1.png

データはリアルタイムに更新される

便利!!

##使い方

import tensorboardX as tbx
import torch

#出力先のファイル名指定
writer = tbx.SummaryWriter("runs/exp-1")

for i in range(20):
    writer.add_scalar("group/epoch", torch.randn(1).item(), i)
writer.close()

これを実行すると

2.png

こんな感じのファイルが出来て

これを読み込むために

3.png

コマンド実行

あとはhttp://localhost:6006

にアクセスするだけ!

###注意事項

起動したtensorboardは強制終了するとポートが占有されたままになってしまいます

必ずCTRL + C で終了させましょう

もし間違えてしまったら

$ tensorboard --logdir="path" --port=6007

などしましょう

###よくあるミス
tensorboardを起動するディレクトリと参照先のディレクトリがうまく対応していないとブラウザで見てもINACTIVEになります

tensorboard コマンドを実行したディレクトリからのpathになっているか確認してください(体験談)

###参照
reference
https://tensorboardx.readthedocs.io/en/latest/tutorial.html

17
13
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
17
13

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?