Commit dfafed3b authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

shut down parallel in instrumental correction

parent ba060b49

csst/msc/inst_corr.py

+2 −1
Original line number Diff line number Diff line
@@ -3,6 +3,7 @@ from pathlib import Path 
import numpy as np

from ccdproc import cosmicray_lacosmic

from deepCR import deepCR

from collections import OrderedDict



from ..core.processor import CsstProcessor, CsstProcStatus

from ..msc import CsstMscImgData
@@ -94,7 +95,7 @@ class CsstMscInstrumentProc(CsstProcessor): 
            inpaint_model = 'ACS-WFC-F606W-2-32'

            model = deepCR(clean_model, inpaint_model, device='CPU', hidden=50)

            masked, cleaned = model.clean(

                self.__img, threshold=0.5, inpaint=True, segment=True, patch=256, parallel=True, n_jobs=self.n_jobs)

                self.__img, threshold=0.5, inpaint=True, segment=True, patch=256, parallel=False, n_jobs=self.n_jobs)

        else:

            cleaned, masked = cosmicray_lacosmic(ccd=self.__img,

                                                 sigclip=3.,  # cr_threshold