tensorflow-cpu跑深度学习网络,由于训练的数据量太大,导致内存溢出怎么解决?

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, 输入数据大小尺寸。