首先，安装torchsummary
pip isntall torchsummary下面是demo代码，其中(3, 100, 100)分别为单张图片的通道数、高、宽下面是demo
阅读背景：
【Pytorch】快速输出网络模型信息方法
来源：互联网
分享到：
