diff options
author | Guillaume Klein <guillaume.klein@systrangroup.com> | 2018-12-12 17:19:52 +0300 |
---|---|---|
committer | Guillaume Klein <guillaume.klein@systrangroup.com> | 2018-12-12 17:19:52 +0300 |
commit | b2fb23735ba315a76c7ebb8d62d19afb687ba07b (patch) | |
tree | 8fb2cc17e6d2e2651852ac6e2cc90cabbecb5f0c | |
parent | ac5970513d48eb24e95b563581c7390025b0de86 (diff) |
Improve TensorFlow converter compatibilityv0.10.0
-rw-r--r-- | python/ctranslate2/converters/opennmt_tf.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/python/ctranslate2/converters/opennmt_tf.py b/python/ctranslate2/converters/opennmt_tf.py index dc05fb97..773229aa 100644 --- a/python/ctranslate2/converters/opennmt_tf.py +++ b/python/ctranslate2/converters/opennmt_tf.py @@ -11,7 +11,7 @@ from ctranslate2.specs import catalog def load_model(model_dir, src_vocab=None, tgt_vocab=None): """Loads variables and vocabularies from a TensorFlow checkpoint or SavedModel.""" - if tf.saved_model.maybe_saved_model_directory(model_dir): + if tf.saved_model.loader.maybe_saved_model_directory(model_dir): config = tf.ConfigProto(device_count={'GPU': 0}) with tf.Session(config=config) as sess: meta_graph = tf.saved_model.loader.load(sess, ["serve"], model_dir) |