Pytorch統計網絡參數數量及查看網絡中的參數

#網絡參數數量
def get_parameter_number(net):
    total_num = sum(p.numel() for p in net.parameters())
    trainable_num = sum(p.numel() for p in net.parameters() if p.requires_grad)
    return {'Total': total_num, 'Trainable': trainable_num}

#查看網絡參數
print(model.state_dict())