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