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