diff options
| author | Lester Solbakken <lesters@users.noreply.github.com> | 2021-10-04 13:04:03 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-10-04 13:04:03 +0200 |
| commit | fed386f182a0b600a72b333bb308c15870c3f04e (patch) | |
| tree | 3245287bbc4fb18b42f126172baf37c21fbba1af | |
| parent | 3f68eb7eb2f03020bed40f328294a0c9300fa17e (diff) | |
| parent | e950a67173c1b91583d14fe343376abb25935ce2 (diff) | |
Merge pull request #19413 from vespa-engine/lesters/support-updated-tf2onnx
Support tf2onnx versions >= 1.9.1
| -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); } |