Update model.py

Pawandeep-prog · Oct 7, 2020 · ab3f3d0 · ab3f3d0
1 parent 87fb08f
commit ab3f3d0
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/model.py b/model.py
@@ -22,7 +22,7 @@
 
 dec_embed = embed(dec_inp)
 dec_lstm = LSTM(400, return_sequences=True, return_state=True)
-dec_op, _, _ = enc_lstm(enc_embed)
+dec_op, _, _ = dec_lstm(dec_embed, initial_state=enc_states)
 
 dense = Dense(VOCAB_SIZE, activation='softmax')
 
@@ -33,8 +33,6 @@
 
 
 
-
-
 model.compile(loss='categorical_crossentropy',metrics=['acc'],optimizer='adam')
 
-model.fit([encoder_inp, decoder_inp],decoder_final_output,epochs=4)
+model.fit([encoder_inp, decoder_inp],decoder_final_output,epochs=4)