From 9e4c6db88281e1d489ebe0317ea12241da378703 Mon Sep 17 00:00:00 2001 From: leufen1 <l.leufen@fz-juelich.de> Date: Tue, 18 May 2021 15:27:16 +0200 Subject: [PATCH] add gc collect statement --- mlair/helpers/filter.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/mlair/helpers/filter.py b/mlair/helpers/filter.py index ced279cc..ff3bb6a5 100644 --- a/mlair/helpers/filter.py +++ b/mlair/helpers/filter.py @@ -101,6 +101,7 @@ class ClimateFIRFilter: var_dim=var_dim, plot_index=i, padlen_factor=padlen_factor) filtered.append(fi) h.append(hi) + gc.collect() # calculate residuum input_data = input_data - fi @@ -396,13 +397,13 @@ class ClimateFIRFilter: # filt = xr.apply_ufunc(fir_filter_vectorized, filter_input_data, time_axis, # input_core_dims=[[new_dim], []], output_core_dims=[[new_dim]], vectorize=True, # kwargs=kwargs) - with TimeTracking(name="convolve"): - slicer = slice(int(-(length - 1) / 2), int((length - 1) / 2)) - filt = xr.apply_ufunc(fir_filter_convolve_vectorized, filter_input_data.sel({new_dim: slicer}), - input_core_dims=[[new_dim]], - output_core_dims=[[new_dim]], - vectorize=True, - kwargs={"h": h}) + # with TimeTracking(name="convolve"): + slicer = slice(int(-(length - 1) / 2), int((length - 1) / 2)) + filt = xr.apply_ufunc(fir_filter_convolve_vectorized, filter_input_data.sel({new_dim: slicer}), + input_core_dims=[[new_dim]], + output_core_dims=[[new_dim]], + vectorize=True, + kwargs={"h": h}) # plot if self.plot_path is not None: -- GitLab