0

I have a model trained in CUDNNLSTM, how can I use this in CPU? How can we export CUDNNLSTM variables to CPU bases weights so that the trained model can be run on CPU also.

Deepak Banka
  • 591
  • 1
  • 6
  • 24
  • When I freeze the model into a pb file and run using that I get this error: Not found: Op type not registered 'CudnnRNN' in binary running on . Make sure the Op and Kernel are registered in the binary running in this process – Deepak Banka Apr 24 '18 at 08:50

1 Answers1

0

One safe way, that always works, would be to simply save the variables to some file with appropriate format like hdf5 and manually load them again. This way you have complete control over what is happening and don't depend on the system you are using. You can even train a model in TensorFlow, save the variables and load them for use with a different library like PyTorch (assuming you defined the exact same model there).

Alexander Harnisch
  • 624
  • 3
  • 8