File tree Expand file tree Collapse file tree 1 file changed +14
-8
lines changed 
captum/optim/models/_image Expand file tree Collapse file tree 1 file changed +14
-8
lines changed Original file line number Diff line number Diff line change @@ -291,14 +291,20 @@ def __init__(
291291        """ 
292292        Args: 
293293
294-             in_channels (int, optional): The number of input channels to use for the 
295-                 inception module. 
296-             c1x1 (int, optional): 
297-             c3x3reduce (int, optional): 
298-             c3x3 (int, optional): 
299-             c5x5reduce (int, optional): 
300-             c5x5 (int, optional): 
301-             pool_proj (int, optional): 
294+             in_channels (int): The number of input channels to use for the first 
295+                 layers of the inception module branches. 
296+             c1x1 (int): The number of output channels to use for the first layer in 
297+                 the c1x1 branch. 
298+             c3x3reduce (int): The number of output channels to use for the first layer 
299+                 in the c3x3 branch. 
300+             c3x3 (int): The number of output channels to use for the second layer in 
301+                 the c3x3 branch. 
302+             c5x5reduce (int): The number of output channels to use for the first layer 
303+                 in the c5x5 branch. 
304+             c5x5 (int): The number of output channels to use for the second layer in 
305+                 the c5x5 branch. 
306+             pool_proj (int): The number of output channels to use for the second layer 
307+                 in the pool branch. 
302308            activ (type of nn.Module, optional): The nn.Module class type to use for 
303309                activation layers. 
304310                Default: ``nn.ReLU`` 
    
 
   
 
     
   
   
          
     
  
    
     
 
    
      
     
 
     
    You can’t perform that action at this time.
  
 
    
  
     
    
      
        
     
 
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments