CODEWITHSUNDEEP
pythontensorflowkeraskeras-layercudnn
Ryan
Ryan

Reputation: 11

Tensorflow CuDNNLSTM keras error. TypeError: ('Keyword argument not understood:', 'time_major')

After I save a CuDNNLSTM model in tensorflow 1.13.1, I try to load it back immediately to use it.

I get the following error

TypeError: ('Keyword argument not understood:', 'time_major').

I looked through the source code for the tensorflow for python and it seems that the _init__() for the base class _CuDNNRNN() has completely removed time_major from being initialized.
Then why was time_major written to file with model.save()?

Upvotes: 1

Views: 2232

Answers (1)

dr. atom
dr. atom

Reputation: 53

Changing the import

from keras.models import model_from_json

to

from tensorflow.keras.models import model_from_json

helped in my situation.

Upvotes: 3

Related Questions