File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff 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 ]
Original file line number Diff line number Diff 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 (
You can’t perform that action at this time.
0 commit comments