TensorFlow 模型建立与训练

大神们,调用 MobileNetV2 这一节中,示例的代码没有改变,运行报错,帮我看看错在哪儿?报错如下:
tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape [50,96,112,112] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc [Op:FusedBatchNormV3]