SSD.Pytorch icon indicating copy to clipboard operation
SSD.Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon midasklr

Memory Error

Open Sagar1094 opened this issue 5 years ago • 0 comments

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.

Sagar1094 avatar Sep 28 '20 13:09 Sagar1094