Include a generator for batches of jets/tracks during preprocessing to avoid that too large datasets cannot be ladet