Merge pull request #592 from panyx0718/master

Allow softplacement for ResNet

Merge pull request #592 from panyx0718/master
Allow softplacement for ResNet
354025b0 · Xin Pan · GitHub · 199db00e · d93ffd0b · 354025b0
Commit 354025b0 authored Oct 27, 2016 by Xin Pan Committed by GitHub Oct 27, 2016
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

resnet/resnet_main.py resnet/resnet_main.py +2 -1

No files found.
--- a/resnet/resnet_main.py
+++ b/resnet/resnet_main.py
@@ -58,7 +58,8 @@ def train(hps):
                           save_summaries_secs=60,
                           save_model_secs=300,
                           global_step=model.global_step)
-  sess = sv.prepare_or_wait_for_session()
+  sess = sv.prepare_or_wait_for_session(
+      config=tf.ConfigProto(allow_soft_placement=True))

  step = 0
  lrn_rate = 0.1