423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 03:42:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

Downgrade "Assigning param_group " msg to DEBUG

Browse Source 
Summary: <See title>

Reviewed By: bottler

Differential Revision: D41534524

fbshipit-source-id: 9c39198b9b8d5fc95f857b03ad39bfe0bd720cbb
This commit is contained in:
David Novotny 2022-11-28 02:58:15 -08:00 committed by Facebook GitHub Bot
parent c3a6ab02da
commit 35f8cb9430
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
projects/implicitron_trainer/impl/optimizer_factory.py
View File

@ -304,7 +304,7 @@ class ImplicitronOptimizerFactory(OptimizerFactoryBase):
for name, param in module.named_parameters(recurse=False): for name, param in module.named_parameters(recurse=False):
if param.requires_grad: if param.requires_grad:
group_name = mapping.get(name, default_group) 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) param_groups[group_name].append(param)
# If children have defined default param group then use it else pass # If children have defined default param group then use it else pass