摘要: 我之前用keras 调用 tf 后端训练的时候,经常出现显存分配错误的问题。保险的方法是手动指定显存分配。 import tensorflow as tffrom keras.backend.tensorflow_backend import set_sessionconfig = tf.Confi 阅读全文
posted @ 2019-06-04 10:16 yjy888 阅读(1964) 评论(0) 推荐(0) 编辑