Skip to content

Commit

Permalink
Back out "Merge logs torchrec/distributed/planner/stats.py"
Browse files Browse the repository at this point in the history
Summary:
Original commit changeset: 347dfb3e5fd7

Original Phabricator Diff: D48619298

Differential Revision: D49835790
  • Loading branch information
hlhtsang authored and facebook-github-bot committed Oct 2, 2023
1 parent 177af2c commit 7e5d6e2
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions torchrec/distributed/planner/stats.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,6 @@
# LICENSE file in the root directory of this source tree.

import logging
import os
from collections import defaultdict
from typing import Any, cast, Dict, List, Optional, Tuple, Union

Expand Down Expand Up @@ -391,7 +390,8 @@ def log(

self._stats_table.append("#" * self._width)

logger.info(os.linesep + os.linesep.join(self._stats_table))
for row in self._stats_table:
logger.info(row)

def _get_shard_stats(
self,
Expand Down

0 comments on commit 7e5d6e2

Please sign in to comment.