zoukankan      html  css  js  c++  java
  • state_dict、named_parameters、named_children、named_modules

    1、model.state_dict()

    # 默认是遍历 key,所以param_tensor实际上是键

    for param_tensor in model.state_dict():

         print(param_tensor,' ',model.state_dict()[param_tensor].size())

    2、model.named_parameters()

    # 遍历name, param

    for name, param, in model.named_parameters():

    其中,1、2的内容是一样的

    3、named_children

    每一个children可能是一个层(如Linear),也可能是多个层(如Sequential),内部用数字索引。

    4、named_modules

    只可能是一个层

  • 相关阅读:
    mysqldump
    设计模式
    设计模式
    设计模式
    设计模式
    PHP 调试
    PHP 调试
    Windows 下手工搭建 LNMP 环境
    设计模式
    设计模式
  • 原文地址:https://www.cnblogs.com/liujianing/p/13425102.html
Copyright © 2011-2022 走看看