Как мне преобразовать модель оценки тензорного потока 2.0 в облегченный тензор потока? - PullRequest
Новая
       46

Как мне преобразовать модель оценки тензорного потока 2.0 в облегченный тензор потока?

2 голосов
/ 22 октября 2019

Следующий код, который я имею ниже, производит обычную модель тензорного потока, но когда я пытаюсь преобразовать ее в тензор потока облегченный, он не работает, я следовал следующей документации:

https://www.tensorflow.org/tutorials/estimator/linear 1 https://www.tensorflow.org/lite/guide/get_started

export_dir = "tmp"
serving_input_fn = tf.estimator.export.build_parsing_serving_input_receiver_fn(
  tf.feature_column.make_parse_example_spec(feat_cols))

estimator.export_saved_model(export_dir, serving_input_fn)

# Convert the model.
converter = tf.lite.TFLiteConverter.from_saved_model("tmp/1571728920/saved_model.pb")
tflite_model = converter.convert()

Сообщение об ошибке 

Traceback (most recent call last):
  File "C:/Users/Dacorie Smith/PycharmProjects/JamaicaClassOneNotifableModels/ClassOneModels.py", line 208, in <module>
    tflite_model = converter.convert()
  File "C:\Users\Dacorie Smith\PycharmProjects\JamaicaClassOneNotifableModels\venv\lib\site-packages\tensorflow_core\lite\python\lite.py", line 400, in convert
    raise ValueError("This converter can only convert a single "
ValueError: This converter can only convert a single ConcreteFunction. Converting multiple functions is under development.

Извлечение из документации

Конвертер TensorFlow LiteКонвертер TensorFlow Lite - это инструмент, доступный в виде API-интерфейса Python, который преобразует обученные модели TensorFlow в формат TensorFlow Lite, а также может вносить оптимизации, описанные в разделе 4. Оптимизация вашей модели.

В следующем примере показанTensorFlow SavedModel преобразуется в формат TensorFlow Lite:

импортировать тензорный поток как tf

converter = tf.lite.TFLiteConverter.from_saved_model (сохраненный_model_dir) tflite_model = converter.convert () открыть ("преобразованный_).tflite "," wb "). write (tflite_model)

...