Skip to content

Latest commit

 

History

History
32 lines (21 loc) · 814 Bytes

README.md

File metadata and controls

32 lines (21 loc) · 814 Bytes
Raw

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:

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