nurion4hep / KSC-RnD-2024-Transformer4MET

0 stars 4 forks source link

add a delphes dataset and a perceiver-based model #14

Closed slowmoyang closed 1 month ago

slowmoyang commented 1 month ago

there are three major changes

  1. I added a dataset diffmet.data.datasets.DelphesDataset taking nuplised delphes file as input. I generate samples with slowmoyang/KSC-RnD-2024-dataset and make ntuples using slowmoyang/KSC-RnD-2024-ntuple. The repositories don't have instructions yet.

  2. I added a new backbone (diffmet.nn.perceiver.Perceiver). In the PU200 environment, we face O(10^4) PF objects per event, leading to OOM when using Transformer as the backbone. So, I added Perceiver, which is a memory-efficient variant of Transformer. If you are instructed in Perceiver, please take a look at the following links.

  3. I added diffmet.callbacks.ResultWriter creating bias and resolution plots.

slowmoyang commented 1 month ago

tested on neuron