Tensorflow:If you want to see a list of allocated tensors when OOM happens, add report_tensor_alloca

显存不足,将batch_size改小。

posted @ 2019-11-13 14:28  旅人_Eric  阅读(1924)  评论(0编辑  收藏  举报