统计pytoch中常用模型的计算密度

目标：

统计深度学习中常见模型的计算密度
模型包括：MobileNet V2、ResNet、VGG16等

---

学习内容：

使用的工具：
模型：torchvision自带的模型库
统计工具：torchstat
输入：1x3x224x224
关键代码：

import torchvision
from torchstat import stat
net = torchvision.models.resnet101(num_classes=10)
print(net)
stat(net,(3,224,224))

---

结果：

---