Memory Error

[Sagar1094]

Hi,
I am using google colab for training on my custom dataset. I am facing memory error, I have 1800 images in which 1300 are training images, 300 valid and 200 test images. I tried training on 512 input bs=16 and error occurs, so I tried multiple combinations and still getting the error. Combinations tried bs=8, bs=4 with input 512. bs=16,8,4,1 with input 300.

Could you please recommend how much GPU memory should I use to train my model.