small fix

rwth-i6 · Jan 13, 2025 · 927fe75 · 927fe75
1 parent f66f6e3
commit 927fe75
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/users/zeyer/experiments/exp2024_04_23_baselines/ctc_recog_ext.py b/users/zeyer/experiments/exp2024_04_23_baselines/ctc_recog_ext.py
@@ -806,7 +806,7 @@ def finish(self, state: LMState):
     float_bytes = 4
 
     print(f"Memory usage {dev_s} after encoder forward:", " ".join(_collect_mem_stats()))
-    snapshot_start = tracemalloc.take_snapshot()
+    snapshot_start = tracemalloc.take_snapshot() if debug_tracemalloc else None
 
     hyps = []
     scores = []