这里使用VIT代码作为例子:
1:在train.py文件导入SummaryWriter。
from torch.utils.tensorboard import SummaryWriter
2:在代码运行前初始化类,SummaryWriter(),括号内可加入用于保存数据的路径。
3:调用tb_writer,向里面添加可视化变量名字,比如学习率,损失函数等。
注意所加的位置在已经生成这些数据的后面。
4:在运行完之后,在保存数据的路径下打开,会找到一个名字很长的文件。
保持这个界面不要动,同时打开终端里的tensorboard(我在jupter notebook里)。
就就可以看到写入的数据了。