Geant4 Cross Reference |
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