Skip to content

Commit

Permalink
Move update_config_with_dtype_overrides logging to debug (#10602) (#1…
Browse files Browse the repository at this point in the history
…0609)

* Move update_config_with_dtype_overrides logging to debug



* update comment



---------

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
  • Loading branch information
akoumpa authored Sep 26, 2024
1 parent 46b8679 commit 835deb6
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions nemo/lightning/pytorch/plugins/mixed_precision.py
Original file line number Diff line number Diff line change
Expand Up @@ -211,12 +211,12 @@ def update_config_with_dtype_overrides(dtype_config, config):
for field in fields(dtype_config):
if not hasattr(config, field.name):
continue
# If we overwrote a value, throw a warning.
# If we overwrote a value, log a debug message.
old_val = getattr(config, field.name)
new_val = getattr(dtype_config, field.name)
if old_val != new_val:
setattr(config, field.name, new_val)
logging.warning(f"Overwrote {type(config).__name__}.{field.name} {old_val} -> {new_val}")
logging.debug(f"Overwrote {type(config).__name__}.{field.name} {old_val} -> {new_val}")
return config


Expand Down

0 comments on commit 835deb6

Please sign in to comment.