Skip to content

Commit

Permalink
Revert of D48619298
Browse files Browse the repository at this point in the history
Summary:
Folks are complaining about how this new format is much harder to get to the lines we care about. None of the existing solutions are at par with the original experience.

Details here: https://fb.workplace.com/groups/1302733780185462/posts/1765846840540818

Differential Revision: D49850102
  • Loading branch information
aay-gup authored and facebook-github-bot committed Oct 2, 2023
1 parent 9b3e0c0 commit a0bc65b
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 a0bc65b

Please sign in to comment.