获取神经网络模型参数的几种方法

if __name__ == '__main__':
    
    # model 是神经网络模型
    for para in model.state_dict():
        # para is the key
        print(para,'t',model.state_dict()[para].size())

    for para in model.parameters():
        print(para.size())

    for para in model.named_parameters():
        # para[0]:parameter name, para[1]:parameter value
        print(para[0],para[1].size())