Commit 75134452 authored by 赵威's avatar 赵威

tractate train config

parent 68cfa708
......@@ -57,8 +57,8 @@ def main():
session_config = tf.compat.v1.ConfigProto()
session_config.gpu_options.allow_growth = True
session_config.gpu_options.per_process_gpu_memory_fraction = 0.7
session_config.inter_op_parallelism_threads = 1
session_config.intra_op_parallelism_threads = 1
# session_config.inter_op_parallelism_threads = 1
# session_config.intra_op_parallelism_threads = 1
estimator_config = tf.estimator.RunConfig(session_config=session_config)
model = tf.estimator.Estimator(model_fn=esmm_model_fn, params=params, model_dir=model_path, config=estimator_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