在trainer.py 中
session_config = tf.ConfigProto(allow_soft_placement=True,
log_device_placement=False)
下添加:
session_config.gpu_options.allow_growth = True
即可
在trainer.py 中
session_config = tf.ConfigProto(allow_soft_placement=True,
log_device_placement=False)
下添加:
session_config.gpu_options.allow_growth = True
即可