From 720f6bf501415a185333b67f3f4760cb4a4aab5f Mon Sep 17 00:00:00 2001 From: jiangxing001 <1484736931@qq.com> Date: Wed, 27 Dec 2023 08:52:23 +0000 Subject: [PATCH] update TensorFlow/built-in/recommendation/DeepCTR_Series_for_TensorFlow/examples/run_fwfm.py. Signed-off-by: jiangxing001 <1484736931@qq.com> --- .../examples/run_fwfm.py | 20 +++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/TensorFlow/built-in/recommendation/DeepCTR_Series_for_TensorFlow/examples/run_fwfm.py b/TensorFlow/built-in/recommendation/DeepCTR_Series_for_TensorFlow/examples/run_fwfm.py index fb3644034..f97c37546 100644 --- a/TensorFlow/built-in/recommendation/DeepCTR_Series_for_TensorFlow/examples/run_fwfm.py +++ b/TensorFlow/built-in/recommendation/DeepCTR_Series_for_TensorFlow/examples/run_fwfm.py @@ -61,16 +61,16 @@ def main(): custom_op.name = "NpuOptimizer" custom_op.parameter_map["precision_mode"].s = tf.compat.as_bytes(args.precision_mode) - if args.profiling: - custom_op.parameter_map["profiling_mode"].b = True - custom_op.parameter_map["profiling_options"].s = tf.compat.as_bytes( - '{"output":"' + args.profiling_dump_path + '", \ - "training_trace":"on", \ - "task_trace":"on", \ - "aicpu":"on", \ - "aic_metrics":"PipeUtilization",\ - "fp_point":"concatenate_1/concat", \ - "bp_point":"training/Adam/gradients/gradients/AddN_83"}') + #if args.profiling: + # custom_op.parameter_map["profiling_mode"].b = True + # custom_op.parameter_map["profiling_options"].s = tf.compat.as_bytes( + # '{"output":"' + args.profiling_dump_path + '", \ + # "training_trace":"on", \ + # "task_trace":"on", \ + # "aicpu":"on", \ + # "aic_metrics":"PipeUtilization",\ + # "fp_point":"concatenate_1/concat", \ + # "bp_point":"training/Adam/gradients/gradients/AddN_83"}') npu_keras_sess = set_keras_session_npu_config(config=sess_config) data = pd.read_csv(os.path.join(args.data_dir, 'criteo_sample.txt')) -- Gitee