diff --git a/TensorFlow/built-in/recommendation/MMoE_Transformer_ID0390_for_TensorFlow/train/fusion_switch.cfg b/TensorFlow/built-in/recommendation/MMoE_Transformer_ID0390_for_TensorFlow/train/fusion_switch.cfg new file mode 100644 index 0000000000000000000000000000000000000000..57e2c717cb18ba8153cc186f8afbe765030495f4 --- /dev/null +++ b/TensorFlow/built-in/recommendation/MMoE_Transformer_ID0390_for_TensorFlow/train/fusion_switch.cfg @@ -0,0 +1,9 @@ +{ + "Switch":{ + "GraphFusion":{ + "RealDiv2MulsFusionPass":"off" + }, + "UBFusion":{ + } + } +} diff --git a/TensorFlow/built-in/recommendation/MMoE_Transformer_ID0390_for_TensorFlow/train/main.py b/TensorFlow/built-in/recommendation/MMoE_Transformer_ID0390_for_TensorFlow/train/main.py index ae177ccdfefe9a401771f41960ee669fbaa4faad..ff30bfc7432966dec73dd7b4d81080926144c6c7 100644 --- a/TensorFlow/built-in/recommendation/MMoE_Transformer_ID0390_for_TensorFlow/train/main.py +++ b/TensorFlow/built-in/recommendation/MMoE_Transformer_ID0390_for_TensorFlow/train/main.py @@ -254,7 +254,10 @@ if __name__ == '__main__': custom_op.parameter_map["use_off_line"].b = True custom_op.parameter_map["precision_mode"].s = tf.compat.as_bytes('allow_mix_precision') custom_op.parameter_map["dynamic_input"].b = True - custom_op.parameter_map["dynamic_graph_execute_mode"].s = tf.compat.as_bytes("lazy_recompile") + custom_op.parameter_map["dynamic_graph_execute_mode"].s = tf.compat.as_bytes("lazy_recompile") + pwdname = os.getcwd() + swith_file = pwdname+"/train/fusion_switch.cfg" + custom_op.parameter_map["fusion_switch_file"].s = tf.compat.as_bytes(swith_file) if iterations_per_loop_mode: custom_op.parameter_map["iterations_per_loop"].i = 10 sess_config.graph_options.rewrite_options.remapping = RewriterConfig.OFF