ZH: Enable multiprocessing in reconstruction

This commit is contained in:
Eric Teunis de Boone 2023-01-12 13:39:06 +01:00
parent 3c99ac7118
commit 0ac1deff34

View file

@ -11,6 +11,7 @@ from mpl_toolkits.mplot3d import Axes3D # required for projection='3d' on old ma
import numpy as np import numpy as np
from os import path from os import path
import pickle import pickle
import joblib
from earsim import REvent from earsim import REvent
from atmocal import AtmoCal from atmocal import AtmoCal
@ -69,6 +70,7 @@ if __name__ == "__main__":
ev.antennas[i].t += total_clock_time ev.antennas[i].t += total_clock_time
N_X, Xlow, Xhigh = 23, 100, 1200 N_X, Xlow, Xhigh = 23, 100, 1200
with joblib.parallel_backend("loky"):
res = rit.reconstruction(ev, outfile=fig_subdir+'/fig.pdf', slice_outdir=fig_subdir+'/', Xlow=Xlow, N_X=N_X, Xhigh=Xhigh) res = rit.reconstruction(ev, outfile=fig_subdir+'/fig.pdf', slice_outdir=fig_subdir+'/', Xlow=Xlow, N_X=N_X, Xhigh=Xhigh)
## Save a pickle ## Save a pickle