diff options
author | Lester Solbakken <lesters@oath.com> | 2020-09-04 11:11:24 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2020-09-04 11:11:24 +0200 |
commit | 5f96bfdcb4dbf38743fd4c93bad9e3acabbbd3b0 (patch) | |
tree | d28943fbc2d372b131c208eb137a517eee4b78de /model-integration | |
parent | 67caf3b6eee690bcd0c7fc7a7666bd2cf41b8816 (diff) |
Add reason for conversion fail to exception
Diffstat (limited to 'model-integration')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/rankingexpression/importer/tensorflow/TensorFlowImporter.java | 7 |
1 files changed, 5 insertions, 2 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 71b9c66a5c0..a879c24b373 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 @@ -71,6 +71,7 @@ public class TensorFlowImporter extends ModelImporter { try { tempDir = Files.createTempDirectory("tf2onnx"); String convertedPath = tempDir.toString() + File.separatorChar + "converted.onnx"; + String outputOfLastConversionAttempt = ""; for (int opset : onnxOpsetsToTry) { log.info("Converting TensorFlow model '" + modelDir + "' to ONNX with opset " + opset + "..."); Pair<Integer, String> res = convertToOnnx(modelDir, convertedPath, opset); @@ -78,9 +79,11 @@ public class TensorFlowImporter extends ModelImporter { log.info("Conversion to ONNX with opset " + opset + " successful."); return onnxImporter.importModel(modelName, convertedPath); } - log.info("Conversion to ONNX with opset " + opset + " failed. Reason: " + res.getSecond()); + log.fine("Conversion to ONNX with opset " + opset + " failed. Reason: " + res.getSecond()); + outputOfLastConversionAttempt = res.getSecond(); } - throw new IllegalArgumentException("Unable to convert TensorFlow model in '" + modelDir + "' to ONNX."); + throw new IllegalArgumentException("Unable to convert TensorFlow model in '" + modelDir + "' to ONNX. " + + "Reason: " + outputOfLastConversionAttempt); } catch (IOException e) { throw new IllegalArgumentException("Conversion from TensorFlow to ONNX failed for '" + modelDir + "'"); } finally { |