ResourceExhaustedError (see above for traceback): OOM when allocating tensor wit
h shape[2,512,9,9]
[[Node: model_1/conv2d_4/convolution = Conv2D[T=DT_FLOAT, data_format="
NHWC", padding="SAME", strides=[1, 2, 2, 1], use_cudnn_on_gpu=true, _device="/jo
b:localhost/replica:0/task:0/device:GPU:0"](model_1/leaky_re_lu_3/sub, conv2d_4/
kernel/read)]]
[[Node: loss_1/mul/_351 = _Recv[client_terminated=false, recv_device="/
job:localhost/replica:0/task:0/device:CPU:0", send_device="/job:localhost/replic
a:0/task:0/device:GPU:0", send_device_incarnation=1, tensor_name="edge_1601_loss
_1/mul", tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:C
PU:0"]()]]
[5960] Failed to execute script train
咩事
