【深度学习实战（20）】使用torchsummary打印模型结构

本文主要是介绍【深度学习实战（20）】使用torchsummary打印模型结构，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

一、安装torchsummary库

pip install torchsummary

二、代码

import torchvision.models as models
from torchsummary import summarymodel = models.AlexNet()
model.to('cuda')
summary(model,(3,224, 224))

这篇关于【深度学习实战（20）】使用torchsummary打印模型结构的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---