Remove reset memory

pomonam · Jul 5, 2024 · c56e285 · c56e285
1 parent 57442ed
commit c56e285
Showing 1 changed file with 0 additions and 4 deletions.
diff --git 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(