Viewing a single comment thread. View all comments

CKtalon t1_jdhkdgc wrote

Code seems fine unless your batch size is too huge. Try running on CPU and see how much RAM is used and debug from there?

3

Rishh3112 OP t1_jdhks1w wrote

my batch size is just 8. I am running it on CPU and my laptop has 8gb of ram and its running fine there.

0