参考这篇博文,在本地能跑通,但是OOM报错,本地资源不足。
2020-08-21 15:08:33.179000: W tensorflow/core/framework/op_kernel.cc:1401] OP_RE QUIRES failed at mkl_util.h:998 : Resource exhausted: OOM when allocating tensor with shape[32,128,768] and type float on /job:localhost/replica:0/task:0/device :CPU:0 by allocator mklcpu