Batchsize issue

@albanD explained some approaches here in a detailed way.
If I’m not mistaken, you are currently using approach 3, so you might want to switch to the second one to save memory.