Warning, /geant4/examples/extended/parameterisations/Par04/vis_onnx_vae.mac is written in an unsupported language. File is not indexed.
0001 /control/execute common_settings_highgran.mac
0002 # Initialize
0003 /run/initialize
0004
0005 /control/execute common_settings_postInit.mac
0006 /control/execute common_settings_vis.mac
0007
0008 # Fast Simulation
0009 # Inference Setup
0010 ## dimension of the latent vector (encoded vector in a Variational Autoencoder model)
0011 /Par04/inference/setSizeLatentVector 10
0012 ## size of the condition vector (energy, angle and geometry)
0013 /Par04/inference/setSizeConditionVector 4
0014 ## path to the model which is set to download by cmake
0015 /Par04/inference/setModelPathName MLModels/Generator.onnx
0016 ## Set ML model to use (VAE, CaloDiT-2)
0017 /Par04/inference/setModelType VAE
0018 /Par04/inference/setProfileFlag 1
0019 /Par04/inference/setOptimizationFlag 0
0020 /Par04/inference/setInferenceLibrary ONNX
0021 ## set mesh size for inference == mesh size of a full sim that
0022 ## was used for training; it coincides with readout mesh size
0023 /Par04/inference/setSizeOfRhoCells 2.325 mm
0024 /Par04/inference/setSizeOfZCells 3.4 mm
0025 /Par04/inference/setNbOfRhoCells 18
0026 /Par04/inference/setNbOfPhiCells 50
0027 /Par04/inference/setNbOfZCells 45
0028 ## Dynamic readout mesh from particle direction needs to be the first fast sim model!
0029 /param/ActivateModel defineMesh
0030 ## ML fast sim, configured with the inference setup /Par04/inference
0031 /param/ActivateModel inferenceModel
0032 /run/beamOn 1
