BaseCollectiveExecutor::StartAbort Resource exhausted: OOM when allocating tensor with shape[32,256,4301,1] and type float on /job:localhost/replica:0/task:0/device:CPU:0 by allocator cpu
[[{{node Cast}}]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.
调低batchsize, 输入数据大小尺寸。