Memory Usage about tensorflow-vgg

[machrisaa]

Hi @jhoh10, my GPU only has 2GB and at most 1.8GB is available for training. I am afraid that the memory size cannot be reduced. But you may move some calculations to be calculated by the CPU and use the general memory instead.

The bottleneck is in the fully connected layers. In most of my project that using the VGG, when I found the GPU memory issue, I will add with tf.device('/cpu:0'): to the largest layer or even the second largest layer.

Hope that is it is helpful.