Problem with training models using fit

Hi guys,

I’m currently developing and image classificator and for that i started watching the tutorial on tensorflow.org .

I’m trying to see how to train a CNN, but as i use model.fit i receive an OOM. I’ve searched on internet but no ones of the answer that i’ve found helped me.

For give you a bit of context i’m using a mobile gtx 1050 and i suspect that the problem is that it hasn’t enough memory, so i ask you some advice on how to succesfully train my models.