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=(