博客园  :: 首页  :: 新随笔  :: 联系 :: 订阅 订阅  :: 管理

2020年11月27日

摘要: 在训练脚本的前面加上 下面代码: import tensorflow as tf import keras config = tf.ConfigProto() config.gpu_options.allow_growth = True keras.backend.tensorflow_backen 阅读全文

posted @ 2020-11-27 17:16 Alex_bd 阅读(430) 评论(0) 推荐(0)