自学深度学习pytorch——Day3 Tensorboard

---

前言

TensorBoard 是一组用于数据可视化的工具。它包含在流行的开源机器学习库 Tensorflow 中。

---

一、安装TensorBoard

进入pytorch环境输入命令安装：pip install tensorboard

二、TensorBoard 的 SummaryWriter

1. 引入

 from torch.utils.tensorboard import SummaryWriter 

2. 指定文件夹 logs

指定一个文件夹，把创建的事件文件存下来。这是 Summarywriter 的 init 初始化做的事情（如果没有则在项目下自动创建）：

writer = SummaryWriter("logs")