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 vec 7 /Par04/inference/setSizeLatentVector 10 8 ## size of the condition vector (energy, angle 9 /Par04/inference/setSizeConditionVector 4 10 ## path to the model which is set to download 11 /Par04/inference/setModelPathName MLModels/Gen 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 18 ## was used for training; it coincides with re 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 214748364 28 /Par04/inference/cuda/setArenaExtendedStrategy 29 /Par04/inference/cuda/setCudnnConvAlgoSearch D 30 /Par04/inference/cuda/setDoCopyInDefaultStream 31 /Par04/inference/cuda/setCudnnConvUseMaxWorksp 32 33 # Fast Simulation 34 /analysis/setFileName 10GeV_100events_fastsim_ 35 ## dynamically set readout mesh from particle 36 ## needs to be the first fast sim model! 37 /param/ActivateModel defineMesh 38 ## ML fast sim, configured with the inference 39 /param/ActivateModel inferenceModel 40 /run/beamOn 100