显存不足,将batch_size改小。
Tensorflow:If you want to see a list of allocated tensors when OOM happens, add report_tensor_alloca
最新推荐文章于 2022-10-14 09:47:44 发布
显存不足,将batch_size改小。