TensorFlow.js转换器中无法识别的参数

时间:2019-06-06 23:00:42

标签: python tensorflow tensorflow.js tensorflowjs-converter tensorflowjs

我正在使用tf_saved_model保存模型

saver = tf.train.Saver()
saver.restore(session, checkpoint)

tutorial之后。 但是当我尝试将模型转换为tensorflowjs模型时出了点问题

具有tensorflow版本

tensorflowjs_converter -v

tensorflowjs 1.1.2

Dependency versions:
  keras 2.2.4-tf
  tensorflow 2.0.0-dev20190606

and related things
tensorflow                       1.13.1              
tensorflow-estimator             1.13.0              
tensorflow-estimator-2.0-preview 1.14.0.dev2019060600
tensorflow-hub                   0.3.0               
tensorflowjs                     1.1.2    

我尝试过

tensorflowjs_converter \
     --input_format=tf_saved_model \
     --output_node_names='embedding, contextual-lstm, attention-lstm, projection' \
     --saved_model_tags=serve \
     /model \
     /web_model

结果: TensorFlow.js model converters.: error: unrecognized arguments: /model /web_model

另一个问题,output_node_names的真正含义是什么? 谢谢。

1 个答案:

答案 0 :(得分:0)

您忘记了./,应将/model更改为./model