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 11:33:32 +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 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,7 +70,8 @@ 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
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 ## Save a pickle
with open(pickle_fname, 'wb') as fp: with open(pickle_fname, 'wb') as fp: