|
@@ -58,7 +58,8 @@ def train(hps):
|
|
|
save_summaries_secs=60,
|
|
save_summaries_secs=60,
|
|
|
save_model_secs=300,
|
|
save_model_secs=300,
|
|
|
global_step=model.global_step)
|
|
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
|
|
step = 0
|
|
|
lrn_rate = 0.1
|
|
lrn_rate = 0.1
|