Skip to content

Commit

Permalink
torchrec: adjust log level for a verbose log line (#1629)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #1629

Avoid spammy logs like this: https://www.internalfb.com/phabricator/paste/view/P1001268340

Reviewed By: henrylhtsang

Differential Revision: D52663009

fbshipit-source-id: ac05e54c2abb3742eb1af14e168d667c52a939aa
  • Loading branch information
shuyi authored and facebook-github-bot committed Jan 17, 2024
1 parent ee19a88 commit 69d6773
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchrec/distributed/planner/partitioners.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,7 +80,7 @@ def _group_and_sort_non_uniform_sharding_options(
if path not in param_count:
param_count[path] = 0
param_count[path] += 1
logger.info(f"param_count is {param_count}")
logger.debug(f"param_count is {param_count}")

sharding_option_groups_by_dependency = {}
for sharding_option in sharding_options:
Expand Down

0 comments on commit 69d6773

Please sign in to comment.