From 12117a11f5228d6b3d75d61cdea8fbf1e4a9d482 Mon Sep 17 00:00:00 2001 From: dachengx Date: Thu, 14 Nov 2024 01:39:04 -0600 Subject: [PATCH] Garbage collection after calculated each MLP chunk --- straxen/plugins/peaks/peak_positions_mlp.py | 1 + 1 file changed, 1 insertion(+) diff --git a/straxen/plugins/peaks/peak_positions_mlp.py b/straxen/plugins/peaks/peak_positions_mlp.py index df77c7071..8cbcd9204 100644 --- 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=(