TensorFlow toco - 从* .pb转换为* .tflite错误

时间:2018-05-11 17:52:10

标签: android tensorflow tensorflow-serving tensorflow-lite

当尝试使用TensorFlow的toco转换器将* .pb模型转换为* .tflite(用于在Android上运行)时,我得到以下内容

2018-05-11 20:38:10.841641: I tensorflow/cc/saved_model/loader.cc:242] Loading SavedModel with tags: { serve }; from: /home/user_name/Downloads/model_directory
2018-05-11 20:38:10.847390: I tensorflow/cc/saved_model/loader.cc:291] SavedModel load for tags { serve }; Status: fail. Took 6042 microseconds.
2018-05-11 20:38:10.848098: F tensorflow/contrib/lite/toco/toco_saved_model.cc:50] **Non-OK-status:** tensorflow::LoadSavedModel(tensorflow::SessionOptions(), tensorflow::RunOptions(), model_path, tags, bundle) **status: Not found:** Could not find meta graph def matching supplied tags: { serve }. To inspect available tag-sets in the SavedModel, please use the SavedModel CLI: `saved_model_cli`Failed to load exported model from /home/user_name/Downloads/model_directory. **Ensure the model contains the required tags 'serve'.**

知道怎么解决吗?

机器细节: 操作系统平台和分发 - ubuntu x64, python版本是3.5.2, TensorFlow从 - pip安装, TensorFlow版本 - cpu版本1.8.0, Bazel版 - 0.13.0, CUDA / cuDNN版本 - 没有cuda, GPU模型和内存 - 没有gpu, 准确的命令来重现 - 不需要

0 个答案:

没有答案
相关问题