HadoopIt / rnn-nlu

A TensorFlow implementation of Recurrent Neural Networks for Sequence Classification and Sequence Labeling
483 stars 171 forks source link

load into tensorflow.js #21

Open nalbion opened 6 years ago

nalbion commented 6 years ago

I'm trying to convert and load the model into tensorflow.js

#TODO --output_node_names='model_tmp/checkpoint' \
tensorflowjs_converter \
    --input_format=tf_saved_model \
    --saved_model_tags=serve \
    model_tmp/model.ckpt-500 \
    model_web

https://github.com/OpenASR/rnn-nlu/blob/master/scripts/convert

When I run the script above I get an error:

IOError: SavedModel file does not exist at: model_tmp/model.ckpt-500

I've tried model_tmp, model_tmp/checkpoint but neither of them work either.

Also, I'm not sure what to provide at output_node_names

https://github.com/tensorflow/tfjs-converter