diff options
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 { |