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

3、named_children

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

4、named_modules

只可能是一个层

你可能感兴趣的:(state_dict、named_parameters、named_children、named_modules)