pytorch使用tensorboard

这里使用VIT代码作为例子：
1：在train.py文件导入SummaryWriter。

from torch.utils.tensorboard import SummaryWriter

2：在代码运行前初始化类，SummaryWriter(),括号内可加入用于保存数据的路径。

3：调用tb_writer,向里面添加可视化变量名字，比如学习率，损失函数等。
注意所加的位置在已经生成这些数据的后面。

4：在运行完之后，在保存数据的路径下打开，会找到一个名字很长的文件。

保持这个界面不要动，同时打开终端里的tensorboard(我在jupter notebook里)。

就就可以看到写入的数据了。