google colaboratory `ResourceExhaustedError` with GPU
问题 I'm trying to fine-tune a Vgg16 model using colaboratory but I ran into this error when training with the GPU. OOM when allocating tensor of shape [7,7,512,4096] INFO:tensorflow:Error reported to Coordinator: <class 'tensorflow.python.framework.errors_impl.ResourceExhaustedError'>, OOM when allocating tensor of shape [7,7,512,4096] and type float [[Node: vgg_16/fc6/weights/Momentum/Initializer/zeros = Const[_class=["loc:@vgg_16/fc6/weights"], dtype=DT_FLOAT, value=Tensor<type: float shape: [7