pytorch打印模型信息——torchinfo

最新推荐文章于 2025-03-29 17:55:32 发布

wxyczhyza

最新推荐文章于 2025-03-29 17:55:32 发布

阅读量1.3k

点赞数

分类专栏： # pytorch 文章标签： pytorch 深度学习人工智能

本文链接：https://blog.youkuaiyun.com/wxyczhyza/article/details/128118418

版权

pytorch 专栏收录该内容

5 篇文章

订阅专栏

这篇博客展示了如何利用torchinfo库的summary函数来获取预训练的ResNet50模型的信息。通过调用summary函数并指定输入尺寸(3, 224, 224)，不包含batch维度，可以查看模型的输入大小、输出大小、参数数量、卷积核大小以及乘加操作次数等关键信息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

用torchinfo库的sumary函数可以打印模型信息，示例如下。注意输入张量中不需要batch维度。

import numpy as np

import torch
import torchvision
import torchinfo

model = torchvision.models.resnet50(pretrained=True)
torchinfo.summary(model, (3, 224, 224), batch_dim=0,
                  col_names=('input_size', 'output_size', 'num_params', 'kernel_size', 'mult_adds'), verbose=1
        )