diff --git a/TensorFlow/built-in/cv/image_synthesis/ViDeNN_ID1146_for_TensorFlow/main_spatialCNN.py b/TensorFlow/built-in/cv/image_synthesis/ViDeNN_ID1146_for_TensorFlow/main_spatialCNN.py index 45c0c428db8ece0ccc3fcb990d347ecd26e9e75c..c777d09f270a71d986fa13a57b7fc2f59055881f 100644 --- a/TensorFlow/built-in/cv/image_synthesis/ViDeNN_ID1146_for_TensorFlow/main_spatialCNN.py +++ b/TensorFlow/built-in/cv/image_synthesis/ViDeNN_ID1146_for_TensorFlow/main_spatialCNN.py @@ -115,6 +115,7 @@ def main(_): custom_op_ = global_config_static.graph_options.rewrite_options.custom_optimizers.add() custom_op_.name = 'NpuOptimizer' custom_op_.parameter_map["use_off_line"].b = True + custom_op_.parameter_map["graph_run_mode"].i = 0 global_config_static.graph_options.rewrite_options.remapping = RewriterConfig.OFF #npu modify end