Skip to content

Commit 1deab13

Browse files
felipemello1Felipe Mello
andauthored
easy - [verbosity] - Reduce log verbosity (#610)
Co-authored-by: Felipe Mello <felipemello@fb.com>
1 parent feb4865 commit 1deab13

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

apps/grpo/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -416,7 +416,7 @@ async def main(cfg: DictConfig):
416416
target_language="ja",
417417
tag="思考",
418418
match_reward=2.0,
419-
debug=True,
419+
debug=False, # set to true for verbose logging
420420
debug_sample_rate=0.1,
421421
), # Japanese language reward with debug
422422
]

src/forge/observability/perf_tracker.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,7 @@ def _start_memory_tracking(self) -> None:
186186

187187
if should_track:
188188
_set_memory_active(True)
189-
torch.cuda.reset_max_memory_allocated()
189+
torch.cuda.reset_peak_memory_stats()
190190
self._start_mem = torch.cuda.memory_allocated()
191191
self._memory_started = True
192192

@@ -202,7 +202,7 @@ def _stop_memory_tracking(self) -> None:
202202
)
203203
record_metric(f"{self.prefix}/memory_peak_max_gb", peak_mem, Reduce.MAX)
204204
_set_memory_active(False)
205-
torch.cuda.reset_max_memory_allocated()
205+
torch.cuda.reset_peak_memory_stats()
206206
self._memory_started = False
207207

208208
def _record_timing_metrics(

0 commit comments

Comments
 (0)