Garbage collection after calculated each MLP chunk (#1467)

XENONnT · Nov 14, 2024 · 8fedc98 · 8fedc98
1 parent f1bb229
commit 8fedc98
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/straxen/plugins/peaks/peak_positions_mlp.py b/straxen/plugins/peaks/peak_positions_mlp.py
@@ -12,6 +12,7 @@ class PeakPositionsMLP(PeakPositionsBaseNT):
 
     provides = "peak_positions_mlp"
     algorithm = "mlp"
+    gc_collect_after_compute = True
 
     tf_model_mlp = straxen.URLConfig(
         default=(