diff --git a/projects/implicitron_trainer/impl/optimizer_factory.py b/projects/implicitron_trainer/impl/optimizer_factory.py index aafe075b..a4f0c0f8 100644 --- a/projects/implicitron_trainer/impl/optimizer_factory.py +++ b/projects/implicitron_trainer/impl/optimizer_factory.py @@ -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