pytorch------>>>统计模型参数量

total_num = sum(p.numel() for p in model.parameters())

trainable_num = sum(p.numel() for p in model.parameters() if p.requires_grad)

print('Total ',total_num)

print('Trainable ',trainable_num)