def count_parameters(model):  # 传入的是模型实例对象
    params = [p.numel() for p in model.parameters() if p.requires_grad]
    for item in params:
        print(f'{
      
      item:>16}')   # 参数大于16的展示
    print(f'________\n{
      
      sum(params):>16}')  # 大于16的进行统计，可以自行修改
count_parameters(net)