Skip to content

Commit d93ffd0

Browse files
author
Xin Pan
committed
Allow softplacement for ResNet
1 parent 6515a41 commit d93ffd0

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

resnet/resnet_main.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,8 @@ def train(hps):
5858
save_summaries_secs=60,
5959
save_model_secs=300,
6060
global_step=model.global_step)
61-
sess = sv.prepare_or_wait_for_session()
61+
sess = sv.prepare_or_wait_for_session(
62+
config=tf.ConfigProto(allow_soft_placement=True))
6263

6364
step = 0
6465
lrn_rate = 0.1

0 commit comments

Comments
 (0)