尝试通过tensorflowjs_converter

时间:2019-03-30 14:44:59

标签: tensorflow tensorflow-estimator tensorflowjs-converter

我有一个tf.estimator模型

model = tf.estimator.DNNClassifier(hidden_​​units = [5,4],feature_columns = feat_cols,n_classes = 2)

通过

导出
feature_spec = tf.feature_column.make_parse_example_spec(feat_cols)
serving_input_receiver_fn = tf.estimator.export.build_parsing_serving_input_receiver_fn(feature_spec)
export_dir = model.export_savedmodel('export', serving_input_receiver_fn)

我能够通过

在笔记本中加载和使用
predict_fn = tf.contrib.predictor.from_saved_model(export_dir)

当我运行tensorflowjs_converter

tensorflowjs_converter --input_format=tf_saved_model --output_format=tensorflowjs ./1553869899 ./web_model

我知道

ValueError: Unsupported Ops in the model before optimization
ParseExample, AsString

我已经环顾四周了,显然ParseExample和AsString是不受支持的Ops。我正在使用不直接调用ParseExample或AsString的非常原始的代码。不会重写部分tensorflow,这似乎是对不受支持的Ops问题的其他答案所需要的。

问题:有没有解决的办法?我是否需要放弃tf.estimator并通过较低级别的API对其进行编码?有没有其他方法可以导出tf.estimator模型或对其进行转换?

谢谢。

0 个答案:

没有答案