| Geant4 Cross Reference |
1 # examplePar04_onnx.mac 1 # examplePar04_onnx.mac
2 # 2 #
3 /control/execute common_settings.mac << 3 # Detector Construction
>> 4 /Par04/detector/setDetectorInnerRadius 80 cm
>> 5 /Par04/detector/setDetectorLength 2 m
>> 6 /Par04/detector/setNbOfLayers 90
>> 7 /Par04/detector/setAbsorber 0 G4_W 1.4 mm true
>> 8 /Par04/detector/setAbsorber 1 G4_Si 0.3 mm true
>> 9 ## 2.325 mm of tungsten =~ 0.25 * 9.327 mm = 0.25 * R_Moliere
>> 10 /Par04/mesh/setSizeOfRhoCells 2.325 mm
>> 11 ## 2 * 1.4 mm of tungsten =~ 0.65 X_0
>> 12 /Par04/mesh/setSizeOfZCells 3.4 mm
>> 13 /Par04/mesh/setNbOfRhoCells 18
>> 14 /Par04/mesh/setNbOfPhiCells 50
>> 15 /Par04/mesh/setNbOfZCells 45
>> 16
>> 17 # Initialize
>> 18 /run/initialize
>> 19
>> 20 /gun/energy 10 GeV
>> 21 /gun/position 0 0 0
>> 22 /gun/direction 0 1 0
4 23
5 # Inference Setup 24 # Inference Setup
6 ## dimension of the latent vector (encoded vec 25 ## dimension of the latent vector (encoded vector in a Variational Autoencoder model)
7 /Par04/inference/setSizeLatentVector 10 26 /Par04/inference/setSizeLatentVector 10
8 ## size of the condition vector (energy, angle 27 ## size of the condition vector (energy, angle and geometry)
9 /Par04/inference/setSizeConditionVector 4 28 /Par04/inference/setSizeConditionVector 4
10 ## path to the model which is set to download 29 ## path to the model which is set to download by cmake
11 /Par04/inference/setModelPathName MLModels/Gen 30 /Par04/inference/setModelPathName MLModels/Generator.onnx
12 /Par04/inference/setProfileFlag 1 31 /Par04/inference/setProfileFlag 1
13 /Par04/inference/setOptimizationFlag 0 32 /Par04/inference/setOptimizationFlag 0
14 ## cuda flag <<
15 /Par04/inference/setCudaFlag 0 <<
16 /Par04/inference/setInferenceLibrary ONNX 33 /Par04/inference/setInferenceLibrary ONNX
17 ## set mesh size for inference == mesh size of 34 ## set mesh size for inference == mesh size of a full sim that
18 ## was used for training; it coincides with re 35 ## was used for training; it coincides with readout mesh size
19 /Par04/inference/setSizeOfRhoCells 2.325 mm 36 /Par04/inference/setSizeOfRhoCells 2.325 mm
20 /Par04/inference/setSizeOfZCells 3.4 mm 37 /Par04/inference/setSizeOfZCells 3.4 mm
21 /Par04/inference/setNbOfRhoCells 18 38 /Par04/inference/setNbOfRhoCells 18
22 /Par04/inference/setNbOfPhiCells 50 39 /Par04/inference/setNbOfPhiCells 50
23 /Par04/inference/setNbOfZCells 45 40 /Par04/inference/setNbOfZCells 45
24 <<
25 # cuda options <<
26 /Par04/inference/cuda/setDeviceId 0 <<
27 /Par04/inference/cuda/setGpuMemLimit 214748364 <<
28 /Par04/inference/cuda/setArenaExtendedStrategy <<
29 /Par04/inference/cuda/setCudnnConvAlgoSearch D <<
30 /Par04/inference/cuda/setDoCopyInDefaultStream <<
31 /Par04/inference/cuda/setCudnnConvUseMaxWorksp <<
32 41
33 # Fast Simulation 42 # Fast Simulation
34 /analysis/setFileName 10GeV_100events_fastsim_ 43 /analysis/setFileName 10GeV_100events_fastsim_onnx.root
35 ## dynamically set readout mesh from particle 44 ## dynamically set readout mesh from particle direction
36 ## needs to be the first fast sim model! 45 ## needs to be the first fast sim model!
37 /param/ActivateModel defineMesh 46 /param/ActivateModel defineMesh
38 ## ML fast sim, configured with the inference 47 ## ML fast sim, configured with the inference setup /Par04/inference
39 /param/ActivateModel inferenceModel 48 /param/ActivateModel inferenceModel
40 /run/beamOn 100 49 /run/beamOn 100