add num threads

eda/esmm/Model_pipline/train.py

@@ -325,7 +325,7 @@ def main(_):
         "dropout": FLAGS.dropout,
         "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)
     Estimator = tf.estimator.Estimator(model_fn=model_fn, model_dir=FLAGS.model_dir, params=model_params, config=config)