From 40e8dd05fef071725e58e2e008012cb8ce2f0d00 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E5=AE=9D=E5=BC=9F?= Date: Thu, 5 May 2022 07:59:39 +0000 Subject: [PATCH 1/3] update --- .../recommendation/WideDeep_ID0028_for_TensorFlow/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/TensorFlow/built-in/recommendation/WideDeep_ID0028_for_TensorFlow/train.py b/TensorFlow/built-in/recommendation/WideDeep_ID0028_for_TensorFlow/train.py index 6852b5438..8aca658e4 100644 --- a/TensorFlow/built-in/recommendation/WideDeep_ID0028_for_TensorFlow/train.py +++ b/TensorFlow/built-in/recommendation/WideDeep_ID0028_for_TensorFlow/train.py @@ -383,6 +383,7 @@ if __name__ == '__main__': custom_op.parameter_map["min_group_size"].b = 1 custom_op.parameter_map["precision_mode"].s = tf.compat.as_bytes("allow_mix_precision") custom_op.parameter_map["hcom_parallel"].b = True + custom_op.parameter_map["iterations_per_loop"].i = 10 if args.over_dump is True: print("NPU overflow dump is enabled") -- Gitee From c6c850ccc60574385f2c759c27747ea7b094884b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E5=AE=9D=E5=BC=9F?= Date: Thu, 5 May 2022 08:00:05 +0000 Subject: [PATCH 2/3] update --- .../WideDeep_ID0028_for_TensorFlow/test/train_performance_8p.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/TensorFlow/built-in/recommendation/WideDeep_ID0028_for_TensorFlow/test/train_performance_8p.sh b/TensorFlow/built-in/recommendation/WideDeep_ID0028_for_TensorFlow/test/train_performance_8p.sh index eef86f05e..bd28f0beb 100644 --- a/TensorFlow/built-in/recommendation/WideDeep_ID0028_for_TensorFlow/test/train_performance_8p.sh +++ b/TensorFlow/built-in/recommendation/WideDeep_ID0028_for_TensorFlow/test/train_performance_8p.sh @@ -13,6 +13,7 @@ RANK_ID_START=0 # 数据集路径,保持为空,不需要修改 data_path="" +sed -i "s/iterations_per_loop = 1/iterations_per_loop = 10/g" `grep -rl "iterations_per_loop = 1" ${cur_path}/../configs/config.py` #设置默认日志级别,不需要修改 export ASCEND_GLOBAL_LOG_LEVEL=3 -- Gitee From 7a8618d51d57d81b155f488dd828ff13d64bda48 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E5=AE=9D=E5=BC=9F?= Date: Thu, 5 May 2022 08:00:29 +0000 Subject: [PATCH 3/3] update --- .../WideDeep_ID0028_for_TensorFlow/test/train_performance_1p.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/TensorFlow/built-in/recommendation/WideDeep_ID0028_for_TensorFlow/test/train_performance_1p.sh b/TensorFlow/built-in/recommendation/WideDeep_ID0028_for_TensorFlow/test/train_performance_1p.sh index 747cd8a71..0b229e638 100644 --- a/TensorFlow/built-in/recommendation/WideDeep_ID0028_for_TensorFlow/test/train_performance_1p.sh +++ b/TensorFlow/built-in/recommendation/WideDeep_ID0028_for_TensorFlow/test/train_performance_1p.sh @@ -13,6 +13,7 @@ RANK_ID_START=0 # 数据集路径,保持为空,不需要修改 data_path="" +sed -i "s/iterations_per_loop = 1/iterations_per_loop = 10/g" `grep -rl "iterations_per_loop = 1" ${cur_path}/../configs/config.py` #设置默认日志级别,不需要修改 export ASCEND_GLOBAL_LOG_LEVEL=3 -- Gitee