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,'\t',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

只可能是一个层

posted @ 2020-08-03 10:57  6+0  阅读(338)  评论(0编辑  收藏  举报