Commit c9174c0f authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

GPU configurations

parent bb94730f
......@@ -332,7 +332,8 @@ class Trainer(object):
logger.info("Initializing !!")
config = tf.ConfigProto(log_device_placement=True)
config = tf.ConfigProto(log_device_placement=True,
gpu_options=tf.GPUOptions(per_process_gpu_memory_fraction=0.333))
config.gpu_options.allow_growth = True
# Pickle the architecture to save
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment