Last active
June 25, 2021 04:32
-
-
Save melgor/41e7d9367410b71dfddc33db34cba85f to your computer and use it in GitHub Desktop.
How to restore CUDNNLSTM of TensorFlow at CPU device? So that it could be used in GPU, CPU or Mobile
@SysuJayce I solved it, you should rebuild model and restore ckpt instead of using import_meta_graph
for anyone who had the same issue I did, remember to set time_major=True in stack_bidirectional_dynamic_rnn().
The CudnnLSTM() function is time major, as opposed to batch major. So remember to transpose your inputs and outputs (swap dimensions 0 and 1)
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hi, I'm running into troubles when restore into CudnnCompatibleLSTMCell from CudnnLSTM, I've run the code above on my machine, and it raised an error, so I'm wandering if I'm using the version that do not support this feature. Would you please provide your tensorflow version that can run the code above? I'm currently run on Tensorflow 1.5.