L2LFlows: Generating High-Fidelity 3D Calorimeter Images

This upload contains the datasets used in arXiv:2302.11594. The file g4-showers_950k_10x10_train_val_test.pt contains the 760k training, 95k validation and 95k test showers as well as their incident energies. It should be loaded as follows: - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - import torch list_tensors = torch.load(args.file_path) for (idx, tensor) in enumerate(list_tensors): [showers_train, showers_val, showers_test, inc_energies_train, inc_energies_val, inc_energies_test] = list_tensors - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - The file g4-showers_665k_10x10_test.pt contains 665k additional showers that were used for the classifier scaling studies, in addition to the 95k test showers from the file g4-showers_950k_10x10_train_val_test.pt. It should be loaded as follows: - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - import torch list_tensors = torch.load("g4-showers_950k_10x10_train_val_test.pt") [showers_geant, inc_energies_geant] = list_tensors - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - A detailed description of how the datasets were simulated can be found in the paper.