I don’t quite understand the use case, as you’ve mentioned the batch size is 1.
Which dimension of your input would correspond to the batch size and which shape are you feeding to the model?
If the mentioned shape is fed in each epoch to the model, when does the OOM error occur?