pytorch named_children()和named_parameters()的区别

named_children()主要用于返回神经网打包的第一层的layer名称

named_parameters()主要用于返回神经网打包的每一层的名字

以resnet18为例子,其每一个layer(也就是basic block)由多个 nn.conv/nn.linear/nn.bn/nn.relu组成,但是在 named_children()里,只会返回最外层的basic block,不会往里面深入,但是named_parameters()正好相反,会深入进去一个子layer一个子layer的返回。pytorch named_children()和named_parameters()的区别_第1张图片

 

你可能感兴趣的:(pytorch)