Open
Description
I am getting valuerror on my Juypter Lab and Vscode. This only work on Coblab.
ValueError: Unrecognized keyword arguments passed to Embedding: {'batch_input_shape': [32, None]}
def build_model(vocab_size, embedding_dim, rnn_units, batch_size):
model = tf.keras.Sequential([
tf.keras.layers.Embedding(vocab_size, embedding_dim, batch_input_shape=[batch_size,None]),**
LSTM(rnn_units),
tf.keras.layers.Dense(vocab_size)**
])
return model
model = build_model(len(vocab), embedding_dim=256, rnn_units=1024, batch_size=32)
Metadata
Metadata
Assignees
Labels
No labels