tensorflow.js转换器抛出``不支持dtype int64''

时间:2019-04-22 17:35:50

标签: tensorflow tensorflowjs-converter

我想使用tensorflowjs转换器转换一个saved_model。由于一些向后兼容的限制,我正在使用tensorflowjs == 0.6.5和tensorflow == 1.12.0。

但是我遇到以下错误:

  File ".../.pyenv5/lib/python3.6/site-packages/tensorflowjs/write_weights.py", 
    line 290, in _assert_valid_weight_entry
    data.dtype.name + ' not supported.')
  ValueError: Error dumping weight module_apply_default/Encoder_en
      /KonaTransformer/ClipToMaxLength/Const, 
  dtype int64 not supported.

我正在使用的命令是:

tensorflowjs_converter 
  --input_format=tf_saved_model \
  --output_node_names='label1/predictions/probabilities' \
  --saved_model_tags=serve \
  /Users/macuser/Desktop/models-master/experiments/tf_tfjs_local_model/20/1555947164 \
  model1

如何在此设置中正确编译保存的模型?

0 个答案:

没有答案
相关问题