TensorFlow KeyError: 'RegexReplace' при прогнозировании на DNNClassifier с использованием tf.contrib.predictor.from_saved_model - PullRequest
0 голосов
/ 12 июня 2018

Я пытаюсь восстановить оценщик и прогнозировать его.

Конфигурация системы : CentOS 7 - x64 - Тензор ЦП == 1,5

Вот как ясохранил оценщик DNNClassifier

 def serving_input_receiver_fn():
      inputs = {"embeddings": tf.placeholder(shape=[None], dtype=tf.string)}
      return tf.estimator.export.ServingInputReceiver(inputs, inputs)

 export_dir = estimator.export_savedmodel(
                 export_dir_base='models/run7',
                 serving_input_receiver_fn=serving_input_receiver_fn)

Я перезагрузил его, используя tf.contrib.predictor.from_saved_model следующим образом

estimator = tf.contrib.predictor.from_saved_model('model/1528805269/')

, но затем я получаю ошибку KeyError: 'RegexReplace' в приведенной выше строке

полная ошибка:

 Traceback (most recent call last):
      File "app.py", line 38, in <module>
        load_model()
      File "app.py", line 21, in load_model
        estimator = tf.contrib.predictor.from_saved_model('model/1528805269/')
      File "/var/www/html/my_project/prj_v4/lib/python3.6/site-packages/tensorflow/contrib/predictor/predictor_factories.py", line 129, in from_saved_model
        graph=graph)
      File "/var/www/html/my_project/prj_v4/lib/python3.6/site-packages/tensorflow/contrib/predictor/saved_model_predictor.py", line 156, in __init__
        loader.load(self._session, tags.split(','), export_dir)
      File "/var/www/html/my_project/prj_v4/lib/python3.6/site-packages/tensorflow/python/saved_model/loader_impl.py", line 216, in load
        saver = tf_saver.import_meta_graph(meta_graph_def_to_load, **saver_kwargs)
      File "/var/www/html/my_project/prj_v4/lib/python3.6/site-packages/tensorflow/python/training/saver.py", line 1838, in import_meta_graph
        **kwargs)
      File "/var/www/html/my_project/prj_v4/lib/python3.6/site-packages/tensorflow/python/framework/meta_graph.py", line 660, in import_scoped_meta_graph
        producer_op_list=producer_op_list)
      File "/var/www/html/my_project/prj_v4/lib/python3.6/site-packages/tensorflow/python/util/deprecation.py", line 316, in new_func
        return func(*args, **kwargs)
      File "/var/www/html/my_project/prj_v4/lib/python3.6/site-packages/tensorflow/python/framework/importer.py", line 433, in import_graph_def
        _RemoveDefaultAttrs(op_dict, producer_op_list, graph_def)
      File "/var/www/html/my_project/prj_v4/lib/python3.6/site-packages/tensorflow/python/framework/importer.py", line 211, in _RemoveDefaultAttrs
        op_def = op_dict[node.op]
KeyError: 'RegexReplace'

Есть предложения, что может быть не так?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...