PyTorchで学習の過程を確認したいときはtensorboardXを使うのが良かったです - Qiita

何かしらのモデルを学習する際に、ロスは順調に下がっているのか等、その経過を逐次確認したいと思うことがよくあります。色々選択肢がありますが、僕が試してきた方法と、その中で良いと思った方法を紹介します。結論はタイトルの通りです。 tensorboardX: https://github.com/lanpa/tensorboard-pytorch 選択肢 まずはじめに、これまで僕がしてきたことは、 学習中には確認しない（学習終了後に、ロス、正解率の推移をmatplotlibなどでプロットする） print(loss) のようなログをイテレーションごとに吐いて、目で推移を確認する (jupyter notebookの場合）bokehのNotebook Handles機能を使って、イテレーション毎にロスのグラフをアップロードしていく 適当なステップ毎に、例えば画像の生成例等を保存していく tenso