小白学视觉 | PyTorch深度学习训练可视化工具tensorboardX

本文来源公众号“小白学视觉”，仅用于学术分享，侵权删，干货满满。

原文链接：PyTorch深度学习训练可视化工具tensorboardX

之前笔者提到了PyTorch的专属可视化工具visdom。但在此之前很多TensorFlow用户更习惯于使用TensorBoard来进行训练的可视化展示。为了能让PyTorch用户也能用上TensorBoard，有开发者提供了PyTorch版本的TensorBoard，也就是tensorboardX。

安装与启动

     熟悉TensorBoard的用户可以无缝对接到tensorboardX，安装方式为：

pip install tensorboardX

     除了要安装PyTorch之外，还需要安装TensorFlow。跟TensorBoard一样，tensorboardX也支持scalar, image, figure, histogram, audio, text, graph, onnx_graph, embedding, pr_curve，video等不同类型对象的可视化展示方式。

tensorboardX和TensorBoard的启动方式一样，直接在终端下运行：

tensorboard --logdir run