Skip to content

Commit

Permalink
removed gc.collect() in ImageDeconvolution
Browse files Browse the repository at this point in the history
  • Loading branch information
hiyoneda committed Jun 14, 2024
1 parent 0067877 commit e0571bb
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions cosipy/image_deconvolution/deconvolution_algorithm_base.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,3 @@
import gc
import numpy as np
import functools
from abc import ABC, abstractmethod
Expand Down Expand Up @@ -142,15 +141,12 @@ def iteration(self):

logger.info("<< Pre-processing >>")
self.pre_processing()
gc.collect()

logger.info("<< E-step >>")
self.Estep()
gc.collect()

logger.info("<< M-step >>")
self.Mstep()
gc.collect()

logger.info("<< Post-processing >>")
self.post_processing()
Expand All @@ -162,8 +158,6 @@ def iteration(self):
stop_iteration = self.check_stopping_criteria()
logger.info("--> {}".format("Stop" if stop_iteration else "Continue"))

gc.collect()

return stop_iteration

def calc_expectation_list(self, model, dict_bkg_norm = None, almost_zero = 1e-12):
Expand Down

0 comments on commit e0571bb

Please sign in to comment.