Browse Source

Remove useless params

tags/v1.2.0-rc1
liuchongming 4 years ago
parent
commit
9446cf0372
1 changed files with 5 additions and 5 deletions
  1. +5
    -5
      mindinsight/mindconverter/graph_based_converter/framework.py

+ 5
- 5
mindinsight/mindconverter/graph_based_converter/framework.py View File

@@ -147,8 +147,8 @@ def tf_installation_validation(func):
type, inner function.
"""

def _f(graph_path: str, sample_shape: tuple, output_folder: str, report_folder: str = None,
input_nodes: str = None, output_nodes: str = None):
def _f(graph_path: str, input_nodes: dict, output_nodes: List[str],
output_folder: str, report_folder: str):
not_integral_error = RuntimeIntegrityError(
f"TensorFlow, "
f"{get_third_part_lib_validation_error_info(['tf2onnx', 'onnx', 'onnxruntime', 'onnxoptimizer'])} "
@@ -171,9 +171,9 @@ def tf_installation_validation(func):
_print_error(not_integral_error)
sys.exit(0)

func(graph_path=graph_path, sample_shape=sample_shape,
output_folder=output_folder, report_folder=report_folder,
input_nodes=input_nodes, output_nodes=output_nodes)
func(graph_path=graph_path,
input_nodes=input_nodes, output_nodes=output_nodes,
output_folder=output_folder, report_folder=report_folder)

return _f



Loading…
Cancel
Save