search

tensorflow / nmt

TensorFlow Neural Machine Translation Tutorial
Apache License 2.0
6.35k stars 1.96k forks source link

How to convert custom tensorflow seq2seq checkpoint model to SavedModel format(pb)? #479

Open bhavikapanara opened 3 years ago

bhavikapanara commented 3 years ago

I have trained my custom seq2seq model that is mostly similar to nmt model where I have added text along with pos and ner of that text as input and output is single text sentence.

now, I have trained checkpoint model ready. but How can I convert that checkpoint to tensorflow SavedModel format(pb)?

I have tried various solution but no success.

Can anybody please help me to solve my problem?

Thnaks, Bhavika

zshancock commented 3 years ago

https://github.com/tensorflow/nmt/issues/294 was a similar issue that cited solutions in https://github.com/tensorflow/serving/issues/712 to resolve.

Hope this helps - my team recently was able to resolve this issue using the above links!