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。