Problem with "time" mode

[AiladReb]

Hello,

When using the 'time' mode, a problem arises in the code because each sample contains a different number of events. This variability in event count prevents the Collate_fn function from concatenating tensors due to mismatched dimensions, resulting in numerous errors in the Flow.py functions.

Do you have a code version where the time mode is used ?

Thank you