diff options
author | Jon Bratseth <bratseth@oath.com> | 2018-08-13 09:52:08 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-08-13 09:52:08 +0200 |
commit | 8420c115a34990d592a287132d5753d8956c4770 (patch) | |
tree | d6df8b74bba2fcde95bd08db1b6982d694f3e6a4 /searchlib | |
parent | 5deabf6d73002692aaaf9e4b4221235f413dc831 (diff) | |
parent | 7ccae75e7f026df580debe58156aa0ee4585714e (diff) |
Merge pull request #6550 from vespa-engine/bratseth/generate-rank-profiles-for-all-models
Bratseth/generate rank profiles for all models
Diffstat (limited to 'searchlib')
-rw-r--r-- | searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/TensorFlowImporter.java | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/TensorFlowImporter.java b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/TensorFlowImporter.java index ff584559a83..303ba228fa6 100644 --- a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/TensorFlowImporter.java +++ b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/TensorFlowImporter.java @@ -23,6 +23,7 @@ public class TensorFlowImporter extends ModelImporter { * @param modelName the name of the model to import, consisting of characters in [A-Za-z0-9_] * @param modelDir the directory containing the TensorFlow model files to import */ + @Override public ImportedModel importModel(String modelName, String modelDir) { try (SavedModelBundle model = SavedModelBundle.load(modelDir, "serve")) { return importModel(modelName, model); |