Hello, and thanks for the answer!
I sadly am not sure what you mean by saying:



 tom:

I moved the batching to the dataset and then just squeezed the singleton additional batch dimension from the system collate


Could you please elaborate a bit?
Thanks in advance.