fix to adjust for #8530 changes (#8612)

2020-11-18 07:25:00 -08:00
parent 2819da02f7
commit cdf1b7ae82
1 changed files with 1 additions and 1 deletions
--- a/examples/seq2seq/finetune.py
+++ b/examples/seq2seq/finetune.py
@@ -148,7 +148,7 @@ class SummarizationModule(BaseTransformer):
            self.save_readable_batch(batch)

        outputs = self(src_ids, attention_mask=src_mask, decoder_input_ids=decoder_input_ids, use_cache=False)
-        lm_logits = outputs[0]
+        lm_logits = outputs["logits"]
        if self.hparams.label_smoothing == 0:
            # Same behavior as modeling_bart.py, besides ignoring pad_token_id
            ce_loss_fct = torch.nn.CrossEntropyLoss(ignore_index=pad_token_id)