Skip to content

F
ffm-baseline
Commit 1a0c0878 authored by Your Name's avatar Your Name
Browse files
Options

add num threads

parent 72834789
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
eda/esmm/Model_pipline/train.py
View file @ 1a0c0878
...@@ -325,7 +325,7 @@ def main(_): ...@@ -325,7 +325,7 @@ def main(_):
"dropout": FLAGS.dropout, "dropout": FLAGS.dropout,
"ctr_task_wgt":FLAGS.ctr_task_wgt "ctr_task_wgt":FLAGS.ctr_task_wgt
} }
config = tf.estimator.RunConfig().replace(session_config = tf.ConfigProto(device_count={'GPU':0, 'CPU':FLAGS.num_threads}), config = tf.estimator.RunConfig().replace(session_config = tf.ConfigProto(device_count={'GPU':0, 'CPU':FLAGS.num_threads},inter_op_parallelism_threads = 1,intra_op_parallelism_threads = 4),
log_step_count_steps=FLAGS.log_steps, save_summary_steps=FLAGS.log_steps) log_step_count_steps=FLAGS.log_steps, save_summary_steps=FLAGS.log_steps)
Estimator = tf.estimator.Estimator(model_fn=model_fn, model_dir=FLAGS.model_dir, params=model_params, config=config) Estimator = tf.estimator.Estimator(model_fn=model_fn, model_dir=FLAGS.model_dir, params=model_params, config=config)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment