Fixed trainer

parent 64acbc93
......@@ -279,7 +279,7 @@ class Trainer(object):
config = tf.ConfigProto(log_device_placement=True)
config.gpu_options.allow_growth = True
with tf.Session(config=True) as session:
with tf.Session(config=config) as session:
tf.initialize_all_variables().run()
......
Markdown is supported
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