GoogLeNet-Inception icon indicating copy to clipboard operation
GoogLeNet-Inception copied to clipboard

Published 20 hours ago • verified publisher icon conan7882

Can not set large batch_size?

Open zhangjinsong3 opened this issue 5 years ago • 1 comments

I trianed the same dataset in caffe-googlenet with the batch size of 32 ,but can only set batch size of 4. That is strange. My device is 1080Ti and has a memory of 12GB.

zhangjinsong3 avatar Apr 03 '19 06:04 zhangjinsong3

I have encountered the same problem. I am using Tesla K80 (12GB memory). My inputs are 256x256 images (456 of them). I can only set the batch size to 2.

jars2007marco avatar Aug 14 '20 18:08 jars2007marco