pytorch中tensorboard的使用

from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter('save') #建立一個保存數據用的東西，save是輸出的文件名

dummy_input = torch.rand(512, 1, 28, 28)  # 網絡中輸入的數據維度
with SummaryWriter(comment='LeNet') as w:
    w.add_graph(net, (dummy_input,))  # net是你的網絡名

 添加完上述代碼後，運行程序後程序裏會出現下列文件夾，如果是在服務器上運行，到服務器上的代碼裏查看是否存在下列文件夾。

  
tensorboard --logdir = C:\Users\huangxin1\PycharmProjects\untitled\runs

TensorBoard 2.6.0 at http://localhost:6006/ (Press CTRL+C to quit)

參考：https://blog.csdn.net/qq_42019881/article/details/124439683