I am trying to train a CNN model using 3 classes and around 500 images in each class. The size of the images are [965*995*1]. When I run the code I get an error saying "Maximum variable size allowed on the device is exceeded."
The model gives an error "Out of memory on device. To view more detail about available memory on the GPU, use 'gpuDevice()'.If the problem persists, reset the GPU by calling 'gpuDevice(1)"when I resize the images to [128 *128].
It runs fine when I resize the images to [64*64].
How can I go around this issue?
Note: The system has 16GB of RAM.
