【常用代码片段】统计模型参数量&查看架构&模型名称

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