我如何将tensorflow 2.0估计器模型转换为tensorflow lite?

问题描述 投票:2回答:1

我下面的以下代码不起作用,根据文档的规定它应该起作用。

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()
python android tensorflow machine-learning tensorflow-estimator
1个回答
0
投票

您收到什么错误?

© www.soinside.com 2019 - 2024. All rights reserved.