Help us understand the problem. What is going on with this article?

tensorboardXの簡単な使い方まとめ

More than 1 year has passed since last update.

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

Why do not you register as a user and use Qiita more conveniently?
  1. We will deliver articles that match you
    By following users and tags, you can catch up information on technical fields that you are interested in as a whole
  2. you can read useful information later efficiently
    By "stocking" the articles you like, you can search right away
Comments
Sign up for free and join this conversation.
If you already have a Qiita account
Why do not you register as a user and use Qiita more conveniently?
You need to log in to use this function. Qiita can be used more conveniently after logging in.
You seem to be reading articles frequently this month. Qiita can be used more conveniently after logging in.
  1. We will deliver articles that match you
    By following users and tags, you can catch up information on technical fields that you are interested in as a whole
  2. you can read useful information later efficiently
    By "stocking" the articles you like, you can search right away