ericteunis/m-thesis-introduction
1
0
Fork 0
mirror of https://gitlab.science.ru.nl/mthesis-edeboone/m-thesis-introduction.git synced 2024-12-22 03:23:34 +01:00
Code Issues Projects Releases Packages Wiki Activity

ZH: Enable multiprocessing in reconstruction

Browse source
This commit is contained in:
Eric Teunis de Boone 2023-01-12 13:39:06 +01:00
parent 3c99ac7118
commit 0ac1deff34
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
simulations/airshower_beacon_simulation/da_reconstruction.py
View file

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