Warning, /geant4/examples/extended/parameterisations/Par04/examplePar04_onnx_calodit.mac is written in an unsupported language. File is not indexed.
0001 # examplePar04_onnx_calodit.mac
0002 #
0003 /control/execute common_settings_lowgran.mac
0004 # Initialize
0005 /run/initialize
0006
0007 /control/execute common_settings_postInit.mac
0008
0009 # Inference Setup
0010 ## path to the model which is set to download by cmake
0011 /Par04/inference/setModelPathName MLModels/cd.onnx
0012 ## Set ML model to use (VAE, CaloDiT-2)
0013 /Par04/inference/setModelType CaloDiT-2
0014 /Par04/inference/setProfileFlag 0
0015 /Par04/inference/setOptimizationFlag 0
0016 ## cuda flag
0017 /Par04/inference/setCudaFlag 0
0018 /Par04/inference/setInferenceLibrary ONNX
0019 ## set mesh size for inference == mesh size of a full sim that
0020 ## was used for training; it coincides with readout mesh size
0021 /Par04/inference/setSizeOfRhoCells 4.65 mm
0022 /Par04/inference/setSizeOfZCells 3.4 mm
0023 /Par04/inference/setNbOfRhoCells 9
0024 /Par04/inference/setNbOfPhiCells 16
0025 /Par04/inference/setNbOfZCells 45
0026
0027 # cuda options
0028 /Par04/inference/cuda/setDeviceId 0
0029 /Par04/inference/cuda/setGpuMemLimit 2147483648
0030 /Par04/inference/cuda/setArenaExtendedStrategy kSameAsRequested
0031 /Par04/inference/cuda/setCudnnConvAlgoSearch DEFAULT
0032 /Par04/inference/cuda/setDoCopyInDefaultStream 1
0033 /Par04/inference/cuda/setCudnnConvUseMaxWorkspace 1
0034
0035 # Fast Simulation
0036 /analysis/setFileName 10GeV_100events_calodit_onnx.root
0037 ## dynamically set readout mesh from particle direction
0038 ## needs to be the first fast sim model!
0039 /param/ActivateModel defineMesh
0040 ## ML fast sim, configured with the inference setup /Par04/inference
0041 /param/ActivateModel inferenceModel
0042 /run/beamOn 100
