引导

1. 安装 torchsummary

在 Anaconda prompt 中进入自己的 pytorch 环境，安装依赖包。

pip install torchsummary

    

    
  • 1

具体如下所示（其中 pytorch-cpu 是我自己的 pytorch 环境）:

torchsummary安装

2. 测试是否下载成功

等待安装完成后运行 python 进入交互式环境，导入 torchsummary, 不报错的话就是安装成功了。

导入torchsummary

3. 输出网络结构

完成以上步骤后，进入自己的 python 编辑环境，运行如下代码。

from torchsummary import summary
from torchvision.models import vgg16  # 以 vgg16 为例

myNet = vgg16() # 实例化网络，可以换成自己的网络
summary(myNet, (3, 64, 64)) # 输出网络结构

  • 1
  • 2
  • 3
  • 4
  • 5

运行代码，结果如下（VGG16）：

在这里插入图片描述
在这里插入图片描述
可以看出，torchsummary 不仅可以查看网络的顺序结构，还有网络参数量，网络模型大小等信息，非常实用。

文章知识点与官方知识档案匹配，可进一步学习相关知识
Python入门技能树人工智能深度学习 262688 人正在系统学习中
# pytorch # python # 深度学习
Logo
乐奇 Rokid 开放社区

更多推荐

用径向菜单完成 macOS 窗口管理的 Loop 上手指引

周三上午，你要同时摆好代码、终端、文档三个窗口。别急着来回拖拽，先按住触发键，把鼠标指向屏幕右侧，窗口就落进了右半屏。这就是 macOS 窗口管理工具 Loop，它用径向菜单和预览窗口，替掉了繁琐的拖边操作。## 🚀 5 步装好 Loop 并完成第一次窗口调整**第 1 步：安装**用 Homebrew 一行装好：```brew install loop```也可以克隆源

avatar 乐奇 Rokid 开放社区

Hydra 1.1 到 1.2 升级指南：@hydra.main() 与 hydra.initialize() 的 config_path 默认行为变更

Hydra 从 1.2 开始，当 `version_base >= "1.2"` 时，`@hydra.main()` 和 `hydra.initialize()` 的 `config_path` 参数默认值由"应用所在的目录"变更为 `None`，即不再向配置搜索路径（Config Search Path）中自动添加任何目录。这篇指南梳理 Hydra 1.0 → 1.1 → 1.2 三个阶段 `c

avatar 乐奇 Rokid 开放社区
cover

苹果缺席的那道题，中国厂商先交了卷

avatar 乐奇 Rokid 开放社区