Updated gpt-2 README

asyml · Mar 1, 2019 · 04a8926 · 04a8926
1 parent 521f7ef
commit 04a8926
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/gpt-2/README.md b/examples/gpt-2/README.md
@@ -139,7 +139,7 @@ output, output_length = decoder(
 output, output_length = decoder(
     memory=source_hidden_states, 
     memory_sequence_length=src_len,
-    inputs=truth_target,
+    inputs=truth_target[:, :-1],
     sequence_length=tgt_len-1,
     decoding_strategy='train_greedy')