Allow softplacement for ResNet

d93ffd0b · Xin Pan · 6515a419 · d93ffd0b
Commit d93ffd0b authored Oct 27, 2016 by Xin Pan
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