Skip to content

GitLab

Commit 7bcb487a authored by Myle Ott's avatar Myle Ott
Browse files

Fix typo

parent 3feaccf3
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
fairseq/criterions/label_smoothed_cross_entropy.py
View file @ 7bcb487a
...@@ -49,7 +49,7 @@ class LabelSmoothedCrossEntropyCriterion(FairseqCriterion): ...@@ -49,7 +49,7 @@ class LabelSmoothedCrossEntropyCriterion(FairseqCriterion):
sample_size = sample['target'].size(0) if self.args.sentence_avg else sample['ntokens'] sample_size = sample['target'].size(0) if self.args.sentence_avg else sample['ntokens']
logging_output = { logging_output = {
'loss': utils.item(loss.data) if reduce else loss.data, 'loss': utils.item(loss.data) if reduce else loss.data,
'nll_loss': utils.item(nll_loss.data) if reduce else loss.data, 'nll_loss': utils.item(nll_loss.data) if reduce else nll_loss.data,
'ntokens': sample['ntokens'], 'ntokens': sample['ntokens'],
'sample_size': sample_size, 'sample_size': sample_size,
} }
......
Markdown is supported
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