cs231 Pytorch tensorboardX 使用

cs231 Pytorch tensorboardX 使用：

tensorboardX支持以下功能： 

Support scalar, image, figure, histogram, audio, text, graph, onnx_graph, embedding, pr_curve and video summaries.

tensorboardX安装部署：

Tested on anaconda2 / anaconda3, with PyTorch 0.4 / torchvision 0.2 / tensorboard 1.7.0

pip install tensorboardX

or build from source:

git clone https://github.com/lanpa/tensorboardX && cd tensorboardX && python setup.py install

requirement for demo_graph.py is tensorboardX>=1.2 and pytorch>=0.4

demo.py代码：

# -*- coding: utf-8 -*-

# 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, 39