Pytorch中named_children()和named_modules()的区别

从定义上讲：

named_children( )

返回包含子模块的迭代器，同时产生模块的名称以及模块本身。

named_modules( )

返回网络中所有模块的迭代器，同时产生模块的名称以及模块本身。

二者返回的都是模型的迭代器，只不过一个返回的是子模块的迭代器，另一个返回的是所有模块的迭代器。

写个程序测试一下：

import torch
import torch.nn as nn

class TestModule(nn.Module):
    def __init__(self):
        super(TestModule,self).__init__()
        self.layer1 = nn.Sequential(
            nn.Conv2d(16,32,3,1),
            nn.ReLU(inplace=True)
        )
        self.layer2 = nn.Sequential(
            nn.Linear(32,10)
        )

    def forward(self,x):
        x = self.layer1(x)
        x = self.layer2(x)

model = TestModule()

for name, module in model.named_children():
    print('children module:', name)

for name, module in model.named_modules():
    print('modules:', name)

>>out:
children module: layer1
children module: layer2
modules: 
modules: layer1
modules: layer1.0
modules: layer1.1
modules: layer2
modules: layer2.0

可以看到named_children只输出了layer1和layer2两个子module，而named_modules输出了包括layer1和layer2下面所有的modolue。