diff --git a/TensorFlow/built-in/nlp/Bert-base_ID0060_for_TensorFlow/src/pretrain/run_pretraining.py b/TensorFlow/built-in/nlp/Bert-base_ID0060_for_TensorFlow/src/pretrain/run_pretraining.py index b2ac9e6ccb3c8da32e3e93d547d02ffe786eac3d..c53666c6cacadea791df748c8d1cfb7b12f08eba 100644 --- a/TensorFlow/built-in/nlp/Bert-base_ID0060_for_TensorFlow/src/pretrain/run_pretraining.py +++ b/TensorFlow/built-in/nlp/Bert-base_ID0060_for_TensorFlow/src/pretrain/run_pretraining.py @@ -629,7 +629,7 @@ def main(_): #run_config = tf.estimator.RunConfig( run_config = NPURunConfig( - dump_config=dump_config + dump_config=dump_config, model_dir=FLAGS.output_dir, save_summary_steps=0, session_config=config,