I am trying to use a simple perceptron to recognize if there is a square or a circle on an image. The images I generated are 300x300 px and I am having issues training the network since the images are large and I seem to run out of memory after a few images.
What would be the correct approach for this?

