pytorch IntermediateLayerGetter

IntermediateLayerGetter

torch是能看源码的，可以直接找。

官方源码：

  

   

    

     

    
    

     

      from collections import OrderedDict
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      from torch import nn
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      class IntermediateLayerGetter(nn.ModuleDict):
     
    
   

    

     

    
    

     
 """
     
    
   

    

     

    
    

     

       Module wrapper that returns intermediate layers from a model
     
    
   

    

     

    
    

     

      
     
    
   

    

     

    
    

     

       It has a strong assumption that the modules have been registered
     
    
   

    

     

    
    

     

       into the model in the same order as they are used.
     
    
   

    

     

    
    

     

       This means that one should **not** reuse the same nn.Module
     
    
   

    

     

    
    

     

       twice in the forward if you want this to work.
     
    
   

    

     

    
    

     

      
     
    
   

    

     

    
    

     

       Additionally, it is only able to query submodules that are directly
     
    
   

    

     

    
    

     

       assigned to the model. So if `model` is passed, `model.feature1` can
     
    
   

    

     

    
    

     

       be returned, but not `model.feature1.layer2`.
     
    
   

    

     

    
    

     

      
     
    
   

    

     

    
    

     

       Arguments:
     
    
   

    

     

    
    

     

       model (nn.Module): model on which we will extract the features
     
    
   

    

     

    
    

     

       return_layers (Dict[name, new_name]): a dict containing the names
     
    
   

    

     

    
    

     

       of the modules for which the 
     
    
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/102475974