嘿,你能搜索到这个问题,说明你说一个认真学习的同学,这个问题的细节值得思考。
欢迎收藏,会持续更新。请仔细看后面的调试界面。
tensorboard打印模型图报错,
除了第一个参数model外,还需要设置第二个参数:input_to_model,也就是输入数据。
如果没有input_to_model,就会报错:
example_kwarg_inputs should be a dict
Error occurs, No graph saved
No graph saved,说明没有图被保存,原因是因为我们没有传入数据,模型没有启动计算流,自然就不会输出模型图。
import tensorboard
from torch.utils.tensorboard import SummaryWriter
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter('runs')
writer.add_graph()
writer.add_graph()
除了model参数要传,input_to_model,这个参数也要传,就是输入一个数据到模型,通过计算后得到模型的结构图。
深度学习神经网络的内容,欢迎关注收藏,会持续更新,如果你遇到什么疑问,也可以评论和留言,大家共同进步。