Model Studio - Out of Memory during training & branches support

Hello,

I have prepared a dataset for use with Model Studio for a time-series prediction.

My input data is around 3GB - 19M rows however I keep getting OoM issues even when memory is bumped to 64GB + A10G GPU.

While I understand that I could reduce the size of my input data, there is no data loader implemented ?

As well, it would be great to support dataset branch as input.

Best regards,