if you see my post above, its months ago, its not actively something I am working on, I didn’t get answer but I solved that time by having smaller batches and less input.
I have few more issues like batch processing, having predictions at the end with a score, do you mind discussing that?