Skip to content

Commit

Permalink
Downgrade "Assigning param_group " msg to DEBUG
Browse files Browse the repository at this point in the history
Summary: <See title>

Reviewed By: bottler

Differential Revision: D41534524

fbshipit-source-id: 9c39198b9b8d5fc95f857b03ad39bfe0bd720cbb
  • Loading branch information
davnov134 authored and facebook-github-bot committed Nov 28, 2022
1 parent c3a6ab0 commit 35f8cb9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion projects/implicitron_trainer/impl/optimizer_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -304,7 +304,7 @@ def traverse(module, default_group: str, mapping: Dict[str, str]) -> None:
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
Expand Down

0 comments on commit 35f8cb9

Please sign in to comment.