Hi there,
I am trying to play with the ImageNet example as here 92. I didn’t change anything of the code. I am using the VGG16 model with a single GPU (1080ti). But the code consistently report out of memory error. Until I reduce the batch size to 50, I can train and test the code. I was wondering is this normal?
BTW, I use cudnn.benchmark=True. My OS is ubuntu 16.04.
Thanks