瀏覽代碼

Merge pull request #592 from panyx0718/master

Allow softplacement for ResNet
Xin Pan 9 年之前
父節點
當前提交
354025b087
共有 1 個文件被更改,包括 2 次插入1 次删除
  1. 2 1
      resnet/resnet_main.py

+ 2 - 1
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