【已解决 含代码调试分析】pytorch,tensorboard.add_graph(),example_kwarg_inputs should be a dictError occurs, No

嘿,你能搜索到这个问题,说明你说一个认真学习的同学,这个问题的细节值得思考。

欢迎收藏,会持续更新。请仔细看后面的调试界面。

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,这个参数也要传,就是输入一个数据到模型,通过计算后得到模型的结构图。

深度学习神经网络的内容,欢迎关注收藏,会持续更新,如果你遇到什么疑问,也可以评论和留言,大家共同进步。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值