Skip to content

S
seminar-breakout
Commit 95f4b9b9 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

clean-up old multi-gpu note

parent 33b212c0
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 6 deletions
example_alexnet.py
View file @ 95f4b9b9
......@@ -73,8 +73,6 @@ def get_model(inputs, is_training):
tf.add_to_collection(COST_VARS_KEY, wd_cost)
add_histogram_summary('.*/W') # monitor histogram of all W
# this won't work with multigpu
#return [prob, nr_wrong], tf.add_n(tf.get_collection(COST_VARS_KEY), name='cost')
return [prob, nr_wrong], tf.add_n([wd_cost, cost], name='cost')
def get_config():
......
example_cifar10.py
View file @ 95f4b9b9
......@@ -79,8 +79,6 @@ def get_model(inputs, is_training):
tf.add_to_collection(COST_VARS_KEY, wd_cost)
add_histogram_summary('.*/W') # monitor histogram of all W
# this won't work with multigpu
#return [prob, nr_wrong], tf.add_n(tf.get_collection(COST_VARS_KEY), name='cost')
return [prob, nr_wrong], tf.add_n([wd_cost, cost], name='cost')
def get_config():
......
example_mnist.py
View file @ 95f4b9b9
......@@ -83,8 +83,6 @@ def get_model(inputs, is_training):
tf.add_to_collection(COST_VARS_KEY, wd_cost)
add_histogram_summary('.*/W') # monitor histogram of all W
# this won't work with multigpu
#return [prob, nr_wrong], tf.add_n(tf.get_collection(COST_VARS_KEY), name='cost')
return [prob, nr_wrong], tf.add_n([wd_cost, cost], name='cost')
def get_config():
......
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