[trick]tensorboardX的使用心得

---

前言

关于tensorboardx 简单使用心得。

---

一、tensorboarx是什么？

tensorboarx是提供实时的深度学习结果的platform。优点是可以及时的查看模型出来的结果，而且对后期调试的时候也很有帮助，比如查看模型的权重分布等等。

二、使用步骤

1.安装

代码如下（示例）：

pip install tensorboardX

也可以从原地址安装：

pip install 'git+https://github.com/lanpa/tensorboardX'

2. 实例 from lanpa/tensorboardX

代码如下（示例）：

# demo.py

import torch
import torchvision.utils as vutils
import numpy as np
import torchvision.models as models
from torchvision import datasets
from tensorboardX import SummaryWriter

resnet18 = models.resnet18(False)
writer = SummaryWriter()
sample_rate = 44100
freqs = [262, 294, 330, 349, 392, 440, 440, 440, 440, 440, 440]

for n_iter in range(100):

    dummy_s1 = torch.