正文
Pytorch在colab和kaggle中使用TensorBoard/TensorboardX可视化
提示：扫一扫查出行【扫一扫了解最新限行尾号】
复制提示
在colab和kaggle内核的Jupyter notebook中如何可视化深度学习模型的参数对于我们分析模型具有很大的意义，相比tensorflow, pytorch缺乏一些的可视化生态包，但是幸好pytorch1.1官方添加了tensorboard的库，可以直接从TORCH.UTILS.TENSORBOARD 调用。
但是就目前的使用体验和反应，还是建议直接使用tensorboardX包在pytorch中进行可视化。
相比本地机中使用notebook进行可视化，在这种远程的notebook中如果使用魔法函数
！tensorboard --logdir
会导致你无法打开tensorboard的http服务（也就是网址）。本文使用一种简单的方法，通过tensorboardX在notebook可视化pytorch模型参数。
步骤：
开始前，先确保安装成功
pip install tensorboardX
pip install tensorboard
pip install tensorflow
第一步import tensorboardX （如何使用tensorboardX请参考reference）：
from tensorboardX import SummaryWriter
第二步初始化一个 SummaryWriter实例：
writer = SummaryWriter()
第三步：
%load_ext tensorboard.notebookor (版本差异，自行选择)
%load_ext tensorboard
第四步把你模型需要在tensorboard中可视化的参数写入writer，每次写入 图像名称，loss/accracy数值， n_iteration
writer.add_scalar('Train/Loss', train_loss, epoch)
writer.add_scalar('Train/Accuracy', train_acc, epoch)
writer.add_scalar('Test/Loss', test_loss, epoch)
writer.add_scalar('Test/Accuracy', test_acc, epoch)
第五步：
%tensorboard --logdir ./
接下来你就可以训练你的模型了，而且在边训练的时候可以边看到模型参数的变化。
