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
只可能是一个层