使用PyTorch的TensorBoard-可视化深度学习指标 | PyTorch系列(二十五)
点击上方 “ AI算法与图像处理 ”,选择加"星标"或“置顶” 重磅干货,第一时间送达 文 |AI_study 原标题: TensorBoard With PyTorch - Visualize Deep Learning Metrics 在本系列的这一点上,我们刚刚完成了训练过程中的网络运行。现在,我们希望获得有关此过程的更多指标,以更好地了解幕后情况。 鸟瞰我们在训练过程中所处的位置。 准备数据 建立模型 训练模型 分析模型的结果 为此使用TensorBoard TensorBoard:TensorFlow的可视化工具包 TensorBoard提供了机器学习实验所需的可视化和工具: 跟踪和可视化指标,例如损失和准确性 可视化模型图(操作和图层) 查看权重,偏差或其他张量随时间变化的直方图 将embedding 投影到较低维度的空间 显示图像,文本和音频数据 分析TensorFlow程序 以及更多 从PyTorch 1.1.0版开始,PyTorch添加了一个tensorboard实用程序包,使我们能够将TensorBoard与PyTorch一起使用。 print ( torch .__version__ ) 1 .1 .0 from torch.utils.tensorboard import SummaryWriter 为PyTorch安装Tensor Board