bug6-ResourceExhaustedError: OOM when allocating tensor with shape[512,15,15,128]
完整错误:tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[512,15,15,128] and type float on /job:localhost/replica:0/task:0 /device:GPU:0 by allocator GPU_0_bfc [Op:Conv2D]解决方法:出现以上类似的错误,主要是因为模型中的batch_