From fc1d2916261b70930dfc29c062042cde51b27e2c Mon Sep 17 00:00:00 2001 From: YueJiang <1206052386@qq.com> Date: Fri, 12 Jan 2024 02:07:09 +0000 Subject: [PATCH] =?UTF-8?q?update=20TensorFlow/built-in/recommendation/Dee?= =?UTF-8?q?pCTR=5FSeries=5Ffor=5FTensorFlow/examples/run=5Fmtl.py.=20?= =?UTF-8?q?=E3=80=90TF=E3=80=91profiling=E9=9C=80=E6=B1=82=E5=81=9A?= =?UTF-8?q?=E4=BA=86=E5=89=8D=E9=83=A8=E6=A0=A1=E9=AA=8C=EF=BC=8C=E4=BB=A3?= =?UTF-8?q?=E7=A0=81=E4=B8=ADprofiling=E9=85=8D=E7=BD=AE=E5=AD=98=E5=9C=A8?= =?UTF-8?q?=E9=97=AE=E9=A2=98=EF=BC=8C=E5=88=A0=E9=99=A4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: YueJiang <1206052386@qq.com> --- .../examples/run_mtl.py | 20 +++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/TensorFlow/built-in/recommendation/DeepCTR_Series_for_TensorFlow/examples/run_mtl.py b/TensorFlow/built-in/recommendation/DeepCTR_Series_for_TensorFlow/examples/run_mtl.py index e20b22363..e2ee2d098 100644 --- a/TensorFlow/built-in/recommendation/DeepCTR_Series_for_TensorFlow/examples/run_mtl.py +++ b/TensorFlow/built-in/recommendation/DeepCTR_Series_for_TensorFlow/examples/run_mtl.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_38"}') + #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_38"}') npu_keras_sess = set_keras_session_npu_config(config=sess_config) column_names = ['age', 'class_worker', 'det_ind_code', 'det_occ_code', 'education', 'wage_per_hour', 'hs_college', -- Gitee