简介
tensorboardX是tensorboard的一个可视化模块, TensorboardX支持scalar, image, figure, histogram, audio, text, graph, onnx_graph, embedding, pr_curve and videosummaries等不同的可视化展示方式,具体介绍移步至项目Github 观看详情。
环境安装
pip install tensorboardX
代码教程
#step1:倒入SummaryWriter
from tensorboardX import SummaryWriter
import numpy as np
#step:2初始化一个实例writer
writer = SummaryWriter('logs')