CaloFlow model checkpoints
Description
These are the model checkpoints for each normalizing flow that was used in "CaloFlow: Fast and Accurate Generation of Calorimeter Showers with Normalizing Flows" and "CaloFlow II: Even Faster and Still Accurate Generation of Calorimeter Showers with Normalizing Flows". The training and evaluation scripts can be found in this git repository.
The training data was based on the same GEANT4 configuration file as the original CaloGAN samples. Said configuration can be found at the CaloGAN repository; the original CaloGAN samples are available at this DOI. The training data can be found at this DOI.
Model checkpoints for each particle (eplus, gamma, piplus) are stored in a separate .tar.gz file. Each tarball contains the following files:
- Flow_I/particle.pt: checkpoint used for flow I.
- Flow_II/particle.pt: checkpoint used for flow II when run as in "CaloFlow: Fast and Accurate Generation of Calorimeter Showers with Normalizing Flows". This is the same as the 'teacher' model of "CaloFlow II: Even Faster and Still Accurate Generation of Calorimeter Showers with Normalizing Flows".
- Flow_II/particle_student.pt: checkpoint used for flow II when run as 'student' model of "CaloFlow II: Even Faster and Still Accurate Generation of Calorimeter Showers with Normalizing Flows".
Each .pt file can be loaded using the functions given in this git repository.
Files
Files
(2.0 GB)
Name | Size | Download all |
---|---|---|
md5:b19290d6ebdbd7241f058285f4ef7aeb
|
681.0 MB | Download |
md5:023ba7a613cb9687e8da30a97850f198
|
681.0 MB | Download |
md5:b9afd912578606f2eeb3aa738b9e5293
|
681.0 MB | Download |
Additional details
Related works
- Is supplement to
- Preprint: https://arxiv.org/abs/2106.05285 (URL)
- Preprint: https://arxiv.org/abs/2110.11377 (URL)
- Journal article: 10.1103/PhysRevD.107.113003 (DOI)
- Journal article: 10.1103/PhysRevD.107.113004 (DOI)
- Dataset: 10.5281/zenodo.7849133 (DOI)