Skip to content

GitLab

Commit 596c9e23 authored by Neal Wu's avatar Neal Wu Committed by GitHub
Browse files

Merge pull request #911 from wookayin/cifar10 

Fix bugs and API usage on cifar10 and cifar10_multi_gpu_train
parents eb62b917 9d96e9fe
Show whitespace changes
Inline Side-by-side
Showing with 20 additions and 19 deletions
tutorials/image/cifar10/cifar10_multi_gpu_train.py
View file @ 596c9e23
...@@ -162,6 +162,7 @@ def train(): ...@@ -162,6 +162,7 @@ def train():
# Calculate the gradients for each model tower. # Calculate the gradients for each model tower.
tower_grads = [] tower_grads = []
with tf.variable_scope(tf.get_variable_scope()):
for i in xrange(FLAGS.num_gpus): for i in xrange(FLAGS.num_gpus):
with tf.device('/gpu:%d' % i): with tf.device('/gpu:%d' % i):
with tf.name_scope('%s_%d' % (cifar10.TOWER_NAME, i)) as scope: with tf.name_scope('%s_%d' % (cifar10.TOWER_NAME, i)) as scope:
......
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