mirror of
				https://github.com/facebookresearch/pytorch3d.git
				synced 2025-11-04 18:02:14 +08:00 
			
		
		
		
	Downgrade "Assigning param_group " msg to DEBUG
Summary: <See title> Reviewed By: bottler Differential Revision: D41534524 fbshipit-source-id: 9c39198b9b8d5fc95f857b03ad39bfe0bd720cbb
This commit is contained in:
		
							parent
							
								
									c3a6ab02da
								
							
						
					
					
						commit
						35f8cb9430
					
				@ -304,7 +304,7 @@ class ImplicitronOptimizerFactory(OptimizerFactoryBase):
 | 
			
		||||
            for name, param in module.named_parameters(recurse=False):
 | 
			
		||||
                if param.requires_grad:
 | 
			
		||||
                    group_name = mapping.get(name, default_group)
 | 
			
		||||
                    logger.info(f"Assigning {name} to param_group {group_name}")
 | 
			
		||||
                    logger.debug(f"Assigning {name} to param_group {group_name}")
 | 
			
		||||
                    param_groups[group_name].append(param)
 | 
			
		||||
 | 
			
		||||
            # If children have defined default param group then use it else pass
 | 
			
		||||
 | 
			
		||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user