OOM error when training data

[wiwengweng]

Hi, all. how is possible to make use of my GTX 950M with only 4G gpu-ram? I tried to modify --epoch 10 --batch_size 10, but still get this error.

ResourceExhaustedError (see above for traceback): OOM when allocating tensor of shape [5,5,2048,4096] and type float
	 [[Node: generator/g_h2/w/Adam/Initializer/zeros = Const[dtype=DT_FLOAT, value=Tensor<type: float shape: [5,5,2048,4096] values: [[[0 0 0]]]...>, _device="/job:localhost/replica:0/task:0/device:GPU:0"]()]]