Update classifier callback and update tokenization example job script by joschkabirk · Pull Request #4 · uhh-pd-ml/omnijet_alpha

joschkabirk · 2025-07-24T14:35:59Z

Summary

This pull request introduces the following changes

logging improvements in classifier callback
move the concatenation of the saved preds and labels to the callback (instead of having it in the lightning module)
Make GPT blocks more configurable

…now done in the callback

joschkabirk added 6 commits July 24, 2025 16:29

Update classifier callback

ecc6a2d

remove label/pred concatenation from lightning module, since this is …

703ffe9

…now done in the callback

Merge branch 'main' of github.com:uhh-pd-ml/omnijet_alpha into updates

52bb3a5

Fix attribute name

45214f7

Fix path

f194112

Update the tokenization example job script

6451edf

joschkabirk changed the title ~~Update classifier callback~~ Update classifier callback and update tokenization example job script Jul 24, 2025

Make GPT blocks more configurable

d4149a1

joschkabirk merged commit 57484ca into main Jul 25, 2025
2 checks passed

joschkabirk deleted the updates branch July 25, 2025 08:16