当前位置:网站首页>torch建立的网络模型使用torchviz显示
torch建立的网络模型使用torchviz显示
2022-07-06 08:27:00 【MAR-Sky】
torchviz.make_dot
MyConvNet = VGG() # 可以是建立的网络
x = torch.randn(4, 3, 256, 256).requires_grad_(True) # 定义一个网络的输入值
y = MyConvNet(x) # 获取网络的预测值
MyConvNetVis = make_dot(y, params=dict(list(MyConvNet.named_parameters()) + [('x', x)]))
MyConvNetVis.format = "png"
# 指定文件生成的文件夹
MyConvNetVis.directory = "data"
# 生成文件
MyConvNetVis.view()
需要注意,输入的格式和网络定义的输入通道格式一样;
依赖graphviz库
网上看到许多需要下载graphviz的zip或mis文件,还配置path;然后用python再进行pip安装后并重启的不知道是什么原理。个人尝试只需要在使用的环境安装graphviz库后就能够使用。
边栏推荐
- 1. Color inversion, logarithmic transformation, gamma transformation source code - miniopencv from zero
- Day29-t77 & t1726-2022-02-13-don't answer by yourself
- Online yaml to CSV tool
- MySQL learning records 12jdbc operation transactions
- [cloud native topic -45]:kubesphere cloud Governance - Introduction and overall architecture of enterprise container platform based on kubernetes
- [2022 Guangdong saim] Lagrange interpolation (multivariate function extreme value divide and conquer NTT)
- JVM 快速入门
- China dihydrolaurenol market forecast and investment strategy report (2022 Edition)
- Pointer advanced --- pointer array, array pointer
- C language - bit segment
猜你喜欢
Fibonacci sequence
[brush questions] top101 must be brushed in the interview of niuke.com
How to use information mechanism to realize process mutual exclusion, process synchronization and precursor relationship
vulnhub hackme: 1
Yyds dry goods inventory three JS source code interpretation eventdispatcher
ROS编译 调用第三方动态库(xxx.so)
What is the use of entering the critical point? How to realize STM32 single chip microcomputer?
[cloud native] teach you how to build ferry open source work order system
Synchronized solves problems caused by sharing
[MySQL] lock
随机推荐
游戏解包的危害及资源加密的重要性
JS pure function
Research and investment forecast report of citronellol industry in China (2022 Edition)
Hill sort c language
[research materials] 2022 China yuancosmos white paper - Download attached
3. File operation 3-with
使用 BR 备份 TiDB 集群数据到兼容 S3 的存储
Roguelike游戏成破解重灾区,如何破局?
使用 TiUP 升级 TiDB
704 二分查找
[brush questions] top101 must be brushed in the interview of niuke.com
VMware virtualization cluster
From monomer structure to microservice architecture, introduction to microservices
pcd转ply后在meshlab无法打开,提示 Error details: Unespected eof
LDAP應用篇(4)Jenkins接入
从 SQL 文件迁移数据到 TiDB
Online yaml to CSV tool
IoT -- 解读物联网四层架构
[research materials] 2021 China online high growth white paper - Download attached
JS inheritance method