From 8fedc98f0e4b365c9792aa7d9e4a7384219f1175 Mon Sep 17 00:00:00 2001 From: Dacheng Xu Date: Thu, 14 Nov 2024 03:30:58 -0500 Subject: [PATCH] Garbage collection after calculated each MLP chunk (#1467) --- 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=(