Skip to content

Fix the gold index issue in fetching logprob#11

Open
fangleai wants to merge 3 commits intoharvardnlp:masterfrom
fangleai:master
Open

Fix the gold index issue in fetching logprob#11
fangleai wants to merge 3 commits intoharvardnlp:masterfrom
fangleai:master

Conversation

@fangleai
Copy link
Copy Markdown

In story generation task, the original code fetches incorrect index of "logprob", leading to incorrect perplexity evaluation.

Originally:
tgt_in = tgt[:-1]
log_probs, attn = ...
...
gold = tgt_in

Edited:
tgt_in = tgt[:-1]
log_probs, attn = ...
...
gold = tgt[1:]

add code sample to evaluate PPL
fix the gold index issue
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant