Problem loading a saved model with biLstm layer

27 views
Skip to first unread message

Rodrigo Pimentel

unread,
May 24, 2018, 9:43:59 AM5/24/18
to TensorFlow.js Discussion
Hi thanks for your help in advance.

If i train and then save the trained model which has a biLSTM layer from the browser, and then try to load the model saved i get an error. Seems like there is a bug with the name of the bilstm layer, 
it currently saves forward_forward_lstm_LSTM1 but should be forward_lstm_LSTM1

This is the error i getn when loading. Unhandled promise rejection Error: Could not find weights in manifest with names:

bidirectional_Bidirectional1/forward_forward_lstm_LSTM1/kernel,
bidirectional_Bidirectional1/forward_forward_lstm_LSTM1/recurrent_kernel,
bidirectional_Bidirectional1/forward_forward_lstm_LSTM1/bias,
bidirectional_Bidirectional1/backward_forward_lstm_LSTM1/kernel,
bidirectional_Bidirectional1/backward_forward_lstm_LSTM1/recurrent_kernel,
bidirectional_Bidirectional1/backward_forward_lstm_LSTM1/bias. 

conv1d_Conv1D6/kernel,
conv1d_Conv1D6/bias,
conv1d_Conv1D4/kernel,
conv1d_Conv1D4/bias,
conv1d_Conv1D5/kernel,
conv1d_Conv1D5/bias,
conv1d_Conv1D7/kernel,
conv1d_Conv1D7/bias,
bidirectional_Bidirectional1/forward_lstm_LSTM1/kernel,
bidirectional_Bidirectional1/forward_lstm_LSTM1/recurrent_kernel,
bidirectional_Bidirectional1/forward_lstm_LSTM1/bias,
bidirectional_Bidirectional1/backward_lstm_LSTM1/kernel,
bidirectional_Bidirectional1/backward_lstm_LSTM1/recurrent_kernel,
bidirectional_Bidirectional1/backward_lstm_LSTM1/bias,
dense_Dense2/kernel,
dense_Dense2/bias.


Shanqing Cai

unread,
May 25, 2018, 10:45:13 AM5/25/18
to TensorFlow.js Discussion
Hi Rodrigo,

Thanks for reporting this issue and sorry for the inconvenience.

I've filed a GitHub issue to track this bug: https://github.com/tensorflow/tfjs/issues/342
Reply all
Reply to author
Forward
0 new messages