diff options
author | Lester Solbakken <lesters@users.noreply.github.com> | 2021-10-04 11:08:09 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-10-04 11:08:09 +0200 |
commit | e950a67173c1b91583d14fe343376abb25935ce2 (patch) | |
tree | 01c4edb51203a8b5c0a0c96672a549d864c1459b /model-integration | |
parent | ba2347b6c4733fbddca32989dae65c0ce8ef939a (diff) |
Support tf2onnx versions >= 1.9.1
Tf2onnx introduced a change to use structured inputs by default from version 1.9.1. Vespa import assumes the actual graph name. The tf2onnx option "--use-graph_names" enables old behavior.
Diffstat (limited to 'model-integration')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/rankingexpression/importer/tensorflow/TensorFlowImporter.java | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/tensorflow/TensorFlowImporter.java b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/tensorflow/TensorFlowImporter.java index 402c6562cc4..0e307992143 100644 --- a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/tensorflow/TensorFlowImporter.java +++ b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/tensorflow/TensorFlowImporter.java @@ -102,7 +102,8 @@ public class TensorFlowImporter extends ModelImporter { private Pair<Integer, String> convertToOnnx(String savedModel, String output, int opset) throws IOException { ProcessExecuter executer = new ProcessExecuter(); - String job = "vespa-convert-tf2onnx --saved-model " + savedModel + " --output " + output + " --opset " + opset; + String job = "vespa-convert-tf2onnx --saved-model " + savedModel + " --output " + output + " --opset " + opset + + " --use-graph-names"; // for backward compatibility with tf2onnx versions < 1.9.1 return executer.exec(job); } |