Skip to content
Snippets Groups Projects
Commit aad2170c authored by Jurica Seva's avatar Jurica Seva
Browse files

Fixed attention mechanism (Mario).

parent 39cc26bc
No related merge requests found
......@@ -197,8 +197,8 @@ def decode_seq(inp_seq):
states_out, states_h, states_c = encoder_model_inf.predict(inp_seq)
states_val = [states_h, states_c]
print('states_val\t', states_val)
input('inference encoder prediction\t')
# print('states_val\t', states_val)
# input('inference encoder prediction\t')
target_seq = np.zeros((1, target_max_sequence_tokenizer))
target_seq[0, 0] = target_vocab['sos']
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment