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):"Initializing !!")
config = tf.ConfigProto(log_device_placement=True)
config = tf.ConfigProto(log_device_placement=True,
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