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

最新推荐文章于 2024-07-01 14:39:46 发布

开飞机的小毛驴儿

最新推荐文章于 2024-07-01 14:39:46 发布

阅读量1.1k

点赞数

分类专栏：深度学习

本文链接：https://blog.csdn.net/jzwong/article/details/108851463

版权

深度学习专栏收录该内容

141 篇文章 18 订阅

订阅专栏

该博客介绍了如何利用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])