Skip to content

关于ghostnetv2模型参数大小的疑问 #286

@uu274

Description

@uu274

请问为什么我在您提供的ghostnetv2的源码中使用summary方法测量模型大小得到的params是19.51m 我使用的代码是
model = ghostnetv2(num_classes=1, width=1.0, dropout=0.2, args=None)
# 模拟输入数据 (batch_size=1, channels=3, height=224, width=224)
input_tensor = torch.randn(1, 3, 224, 224)

# 使用 torchinfo 来打印模型摘要
summary(model, input_data=input_tensor)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions