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

最新推荐文章于 2022-12-30 11:19:49 发布

LUQC638

最新推荐文章于 2022-12-30 11:19:49 发布

阅读量988

点赞数 1

文章标签：神经网络 pytorch 人工智能

本文链接：https://blog.csdn.net/weixin_61445075/article/details/122268883

版权

神经网络模型参数 state_dict named_parameters 深度学习

关键词由CSDN通过智能技术生成


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())