【常见代码片段】 查看模型名称和大小、模块参数数量

    # 自定义
    model = build_model(args).to(device)

    # 查看模型名称和大小
    for name, param in model.ccff.named_parameters():
        print(name, param.size())
    # 查看模块的参数数量
    param_count = sum(p.numel() for p in model.ccff.parameters())
    print(f"Total number of parameters in CCFF: {param_count}")

conv1.conv.weight torch.Size([3584, 3584, 1, 1])
conv1.norm.weight torch.Size([3584])
conv1.norm.bias torch.Size([3584])
conv2.conv.weight torch.Size([3584, 3584, 1, 1])
conv2.norm.weight torch.Size([3584])
conv2.norm.bias torch.Size([3584])
bottlenecks.0.conv1.conv.weight torch.Size([3584, 3584, 3, 3])
bottlenecks.0.conv1.norm.weight torch.Size([3584])
bottlenecks.0.conv1.norm.bias torch.Size([3584])
bottlenecks.0.conv2.conv.weight torch.Size([3584, 3584, 1, 1])
bottlenecks.0.conv2.norm.weight torch.Size([3584])
bottlenecks.0.conv2.norm.bias torch.Size([3584])
bottlenecks.1.conv1.conv.weight torch.Size([3584, 3584, 3, 3])
bottlenecks.1.conv1.norm.weight torch.Size([3584])
bottlenecks.1.conv1.norm.bias torch.Size([3584])
bottlenecks.1.conv2.conv.weight torch.Size([3584, 3584, 1, 1])
bottlenecks.1.conv2.norm.weight torch.Size([3584])
bottlenecks.1.conv2.norm.bias torch.Size([3584])
bottlenecks.2.conv1.conv.weight torch.Size([3584, 3584, 3, 3])
bottlenecks.2.conv1.norm.weight torch.Size([3584])
bottlenecks.2.conv1.norm.bias torch.Size([3584])
bottlenecks.2.conv2.conv.weight torch.Size([3584, 3584, 1, 1])
bottlenecks.2.conv2.norm.weight torch.Size([3584])
bottlenecks.2.conv2.norm.bias torch.Size([3584])
Total number of parameters in CCFF: 411099136