From 3ed2c187db806fb269170b6649a7ff77a75d47e9 Mon Sep 17 00:00:00 2001 From: yongchao1 <297389370@qq.com> Date: Tue, 10 Jan 2023 08:37:41 +0000 Subject: [PATCH 1/2] update BertLarge_ID0634_for_TensorFlow2.X/test/train_performance_8p_192bs_bucket.sh. Signed-off-by: yongchao1 <297389370@qq.com> --- .../test/train_performance_8p_192bs_bucket.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TensorFlow2/built-in/nlp/BertLarge_ID0634_for_TensorFlow2.X/test/train_performance_8p_192bs_bucket.sh b/TensorFlow2/built-in/nlp/BertLarge_ID0634_for_TensorFlow2.X/test/train_performance_8p_192bs_bucket.sh index 225afcb73..bb44c5f9a 100644 --- a/TensorFlow2/built-in/nlp/BertLarge_ID0634_for_TensorFlow2.X/test/train_performance_8p_192bs_bucket.sh +++ b/TensorFlow2/built-in/nlp/BertLarge_ID0634_for_TensorFlow2.X/test/train_performance_8p_192bs_bucket.sh @@ -13,7 +13,7 @@ RANK_ID_START=0 export NPU_ENABLE_PERF=true # 数据集路径,保持为空,不需要修改 data_path="" - +export HCCL_CONNECT_TIMEOUT=1800 #基础参数 需要模型审视修改 #网络名称，同目录名称 Network="BertLarge_ID0634_for_TensorFlow2.X" -- Gitee From aa01c6b3f3c7032cf62bd0cfce62897cc49222b7 Mon Sep 17 00:00:00 2001 From: yongchao1 <297389370@qq.com> Date: Tue, 10 Jan 2023 08:45:10 +0000 Subject: [PATCH 2/2] update BertLarge_ID0634_for_TensorFlow2.X/bert/run_pretraining_bucket.py. Signed-off-by: yongchao1 <297389370@qq.com> --- .../bert/run_pretraining_bucket.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/TensorFlow2/built-in/nlp/BertLarge_ID0634_for_TensorFlow2.X/bert/run_pretraining_bucket.py b/TensorFlow2/built-in/nlp/BertLarge_ID0634_for_TensorFlow2.X/bert/run_pretraining_bucket.py index 2504b92d0..33e064fa3 100644 --- a/TensorFlow2/built-in/nlp/BertLarge_ID0634_for_TensorFlow2.X/bert/run_pretraining_bucket.py +++ b/TensorFlow2/built-in/nlp/BertLarge_ID0634_for_TensorFlow2.X/bert/run_pretraining_bucket.py @@ -163,7 +163,8 @@ def npu_config(): "38,320,38,320,38,320,38,76,38,76,38,76,38,1;" \ "32,384,32,384,32,384,32,76,32,76,32,76,32,1;" \ "28,448,28,448,28,448,28,76,28,76,28,76,28,1;" \ - "24,512,24,512,24,512,24,76,24,76,24,76,24,1" + "24,512,24,512,24,512,24,76,24,76,24,76,24,1;" \ + "27,448,27,448,27,448,27,76,27,76,27,76,27,1" if FLAGS.data_dump_flag: npu_device.global_options().dump_config.enable_dump = True -- Gitee