Loading...
Searching...
No Matches
examplePar04_onnx.mac
Go to the documentation of this file.
1# examplePar04_onnx.mac
2#
3/control/execute common_settings.mac
4
5# Inference Setup
6## dimension of the latent vector (encoded vector in a Variational Autoencoder model)
7/Par04/inference/setSizeLatentVector 10
8## size of the condition vector (energy, angle and geometry)
9/Par04/inference/setSizeConditionVector 4
10## path to the model which is set to download by cmake
11/Par04/inference/setModelPathName MLModels/Generator.onnx
12/Par04/inference/setProfileFlag 1
13/Par04/inference/setOptimizationFlag 0
14## cuda flag
15/Par04/inference/setCudaFlag 0
16/Par04/inference/setInferenceLibrary ONNX
17## set mesh size for inference == mesh size of a full sim that
18## was used for training; it coincides with readout mesh size
19/Par04/inference/setSizeOfRhoCells 2.325 mm
20/Par04/inference/setSizeOfZCells 3.4 mm
21/Par04/inference/setNbOfRhoCells 18
22/Par04/inference/setNbOfPhiCells 50
23/Par04/inference/setNbOfZCells 45
24
25# cuda options
26/Par04/inference/cuda/setDeviceId 0
27/Par04/inference/cuda/setGpuMemLimit 2147483648
28/Par04/inference/cuda/setArenaExtendedStrategy kSameAsRequested
29/Par04/inference/cuda/setCudnnConvAlgoSearch DEFAULT
30/Par04/inference/cuda/setDoCopyInDefaultStream 1
31/Par04/inference/cuda/setCudnnConvUseMaxWorkspace 1
32
33# Fast Simulation
34/analysis/setFileName 10GeV_100events_fastsim_onnx.root
35## dynamically set readout mesh from particle direction
36## needs to be the first fast sim model!
37/param/ActivateModel defineMesh
38## ML fast sim, configured with the inference setup /Par04/inference
39/param/ActivateModel inferenceModel
40/run/beamOn 100

Applications | User Support | Publications | Collaboration