Skip to content

Commit 5ec8765

Browse files
authored
Grammatical updates to GC logging (#1737)
1 parent c9cb304 commit 5ec8765

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchtitan/tools/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,13 +55,13 @@ def run(self, step_count: int):
5555
)
5656
gc.collect()
5757
elif step_count > 1 and step_count % self.gc_freq == 0:
58-
self.collect("Performing periodical GC collection")
58+
self.collect("Performing periodic GC collection")
5959

6060
@staticmethod
6161
def collect(reason: str, generation: int = 1):
6262
begin = time.monotonic()
6363
gc.collect(generation)
64-
logger.info("[GC] %s %.2f seconds", reason, time.monotonic() - begin)
64+
logger.info("[GC] %s took %.2f seconds", reason, time.monotonic() - begin)
6565

6666

6767
# hardcoded BF16 type peak flops for NVIDIA A100, H100, H200, B200 GPU and AMD MI250, MI300X, AMD MI325X and Intel PVC

0 commit comments

Comments
 (0)