CaloDiffusion unofficial repository (WIP)

Pytorch v1.9 was used to implement all models.

Data

Results are presented using the Fast Calorimeter Data Challenge dataset and are available for download on zenodo:

• Dataset 1
• Dataset 2
• Dataset 3

Run the training scripts with

cd scripts
python train_diffu.py  --config CONFIG

• Example configs in [config_dataset1.json/config_dataset2.json/config_dataset3.json]

Sampling with the learned model

python plot.py  --nevts N  --sample  --config CONFIG --model MODEL

Creating the plots shown in the paper

python plot.py  --config CONFIG --model MODEL