search

HEP-KBFI / ml-tau-en-reg

Reconstructing tau leptons using various machine learning techniques
GNU General Public License v3.0
0 stars 1 forks source link

reduce number of layers for ParticleTransformer decay mode classification #23

Closed jpata closed 1 month ago

jpata commented 1 month ago

Made the number of layers of ParticleTransformer configurable, reduce from 8->4. This improves the both the training and validation loss a lot, validation is now decreasing stably as expected:

Screenshot 2024-05-28 at 12 20 13

Now the final plot looks like this: image