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 747cd8a7126cc863e3fbc7e93fd4a47ab9d48e26..0b229e6380930ef6181c5e0fa8f86915d358bb1b 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 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 eef86f05e9107e78f01370fb143a5c4db5bc73ef..bd28f0bebcab54489dfe2768d43819ef223a21d5 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 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 6852b54388f51c8d6e4b3347ebd381fdc23ee41e..8aca658e48dc5359e00fb5e9a27a2c9a61f68ecb 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")