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)