"vscode:/vscode.git/clone" did not exist on "380a0416249b085f7b58baab5fcdd4dc020b5137"
Commit 79e40801 authored by Eli Bixby's avatar Eli Bixby
Browse files

Fix GPUConfig initialization

parent 1b0bf3e6
...@@ -497,12 +497,15 @@ def main(unused_argv): ...@@ -497,12 +497,15 @@ def main(unused_argv):
eval_steps = num_eval_examples // FLAGS.eval_batch_size eval_steps = num_eval_examples // FLAGS.eval_batch_size
# Session configuration. # Session configuration.
sess_config = tf.ConfigProto() sess_config = tf.ConfigProto(
sess_config.allow_soft_placement = True allow_soft_placement=True,
sess_config.log_device_placement = FLAGS.log_device_placement log_device_placement=FLAGS.log_device_placement,
sess_config.intra_op_parallelism_threads = FLAGS.num_intra_threads intra_op_parallelism_threads=FLAGS.num_intra_threads,
sess_config.inter_op_parallelism_threads = FLAGS.num_inter_threads inter_op_parallelism_threads=FLAGS.num_inter_threads,
sess_config.gpu_options.force_gpu_compatible = FLAGS.force_gpu_compatible gpu_options=tf.GPUOptions(
force_gpu_compatible=FLAGS.force_gpu_compatible
)
)
# Hooks that add extra logging that is useful to see the loss more often in # Hooks that add extra logging that is useful to see the loss more often in
# the console as well as examples per second. # the console as well as examples per second.
......
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