From c56e2853d44f6075a1d0f34947700d3032e6b0b9 Mon Sep 17 00:00:00 2001 From: Juhan Bae Date: Fri, 5 Jul 2024 02:15:57 -0400 Subject: [PATCH] Remove reset memory --- kronfluence/computer/factor_computer.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/kronfluence/computer/factor_computer.py b/kronfluence/computer/factor_computer.py index 051fa1a..c59d6fc 100644 --- a/kronfluence/computer/factor_computer.py +++ b/kronfluence/computer/factor_computer.py @@ -298,9 +298,6 @@ def fit_covariance_matrices( total_data_examples=max_partition_examples, ) - self._reset_memory() - - import gc for obj in gc.get_objects(): try: @@ -311,7 +308,6 @@ def fit_covariance_matrices( except: pass - start_time = get_time(state=self.state) with self.profiler.profile("Fit Covariance"): loader = self._get_dataloader(