‘torch._C.Value‘ object has no attribute ‘uniqueName‘神经网络可视化

该博客介绍了如何利用TensorWatch库在不使用Jupyter Notebook或者遇到显示问题时,保存模型结构为PDF文件。作者通过导入自定义的ESPNet_Encoder网络模型,并使用tensorwatch的draw_model方法绘制模型结构,然后调用save方法将网络结构保存到本地,方便离线查看。
摘要由CSDN通过智能技术生成

也有可能是tensorwatch版本过高的原因,我修改为0.8版本即可正常

如果不想用jupyter notebook或者其显示不正常,可以保存在本地查看

structure=tw.draw_model(model,input)

structure.save('network_structure')

会在本地保存一个pdf文件对应网络结构

 

import torch
import tensorwatch as tw
from lanenet_model.blocks import ESPNet_Encoder # 这是我自己定义的一个网络
 
# 其实就两句话
model=ESPNet_Encoder()
tw.draw_model(model, [1, 3, 512, 256])

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值