Import Geant4 11.4.0.beta source tree

examples/extended/parameterisations/Par04/examplePar04_torch_vae.mac

@@ -0,0 +1,35 @@
+#  examplePar04_torch_vae.mac
+#
+
+/control/execute common_settings_highgran.mac
+# Initialize
+/run/initialize
+
+/control/execute common_settings_postInit.mac
+
+# Inference Setup
+## dimension of the latent vector (encoded vector in a Variational Autoencoder model)
+/Par04/inference/setSizeLatentVector 10
+## size of the condition vector (energy, angle and geometry)
+/Par04/inference/setSizeConditionVector 4
+## path to the model which is set to download by cmake
+/Par04/inference/setModelPathName MLModels/Generator.pt
+## Set ML model to use (VAE, CaloDiT-2)
+/Par04/inference/setModelType VAE
+/Par04/inference/setInferenceLibrary TORCH
+## set mesh size for inference == mesh size of a full sim that
+## was used for training; it coincides with readout mesh size
+/Par04/inference/setSizeOfRhoCells 2.325 mm
+/Par04/inference/setSizeOfZCells 3.4 mm
+/Par04/inference/setNbOfRhoCells 18
+/Par04/inference/setNbOfPhiCells 50
+/Par04/inference/setNbOfZCells 45
+
+# Fast Simulation
+/analysis/setFileName 10GeV_100events_vae_torch.root
+## dynamically set readout mesh from particle direction
+## needs to be the first fast sim model!
+/param/ActivateModel defineMesh
+## ML fast sim, configured with the inference setup /Par04/inference
+/param/ActivateModel inferenceModel
+/run/beamOn 100