Warning, /geant4/examples/extended/parameterisations/Par04/examplePar04_onnx.mac is written in an unsupported language. File is not indexed.
0001 # examplePar04_onnx.mac
0002 #
0003 /control/execute common_settings.mac
0004
0005 # Inference Setup
0006 ## dimension of the latent vector (encoded vector in a Variational Autoencoder model)
0007 /Par04/inference/setSizeLatentVector 10
0008 ## size of the condition vector (energy, angle and geometry)
0009 /Par04/inference/setSizeConditionVector 4
0010 ## path to the model which is set to download by cmake
0011 /Par04/inference/setModelPathName MLModels/Generator.onnx
0012 /Par04/inference/setProfileFlag 1
0013 /Par04/inference/setOptimizationFlag 0
0014 ## cuda flag
0015 /Par04/inference/setCudaFlag 0
0016 /Par04/inference/setInferenceLibrary ONNX
0017 ## set mesh size for inference == mesh size of a full sim that
0018 ## was used for training; it coincides with readout mesh size
0019 /Par04/inference/setSizeOfRhoCells 2.325 mm
0020 /Par04/inference/setSizeOfZCells 3.4 mm
0021 /Par04/inference/setNbOfRhoCells 18
0022 /Par04/inference/setNbOfPhiCells 50
0023 /Par04/inference/setNbOfZCells 45
0024
0025 # cuda options
0026 /Par04/inference/cuda/setDeviceId 0
0027 /Par04/inference/cuda/setGpuMemLimit 2147483648
0028 /Par04/inference/cuda/setArenaExtendedStrategy kSameAsRequested
0029 /Par04/inference/cuda/setCudnnConvAlgoSearch DEFAULT
0030 /Par04/inference/cuda/setDoCopyInDefaultStream 1
0031 /Par04/inference/cuda/setCudnnConvUseMaxWorkspace 1
0032
0033 # Fast Simulation
0034 /analysis/setFileName 10GeV_100events_fastsim_onnx.root
0035 ## dynamically set readout mesh from particle direction
0036 ## needs to be the first fast sim model!
0037 /param/ActivateModel defineMesh
0038 ## ML fast sim, configured with the inference setup /Par04/inference
0039 /param/ActivateModel inferenceModel
0040 /run/beamOn 100
