Support input truncation in dataloader
This adds support for reducing e.g. the number of tracks trained on.
The JetDataModule
is also cleaned up.
This adds support for reducing e.g. the number of tracks trained on.
The JetDataModule
is also cleaned up.