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
作者:柒月
Q群 :2122210(嵌入式/机器学习)