[Seq2Seq] Correct import in Seq2Seq Trainer (#8254)

9f1747f9 · Patrick von Platen · GitHub · 504ff7bb · 9f1747f9
Unverified Commit 9f1747f9 authored Nov 03, 2020 by Patrick von Platen Committed by GitHub Nov 03, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 4 deletions

examples/seq2seq/seq2seq_trainer.py examples/seq2seq/seq2seq_trainer.py +1 -4

No files found.
--- a/examples/seq2seq/seq2seq_trainer.py
+++ b/examples/seq2seq/seq2seq_trainer.py
@@ -62,10 +62,7 @@ class Seq2SeqTrainer(Trainer):
            self.loss_fn = torch.nn.CrossEntropyLoss(ignore_index=self.config.pad_token_id)
        else:
            # dynamically import label_smoothed_nll_loss
-            try:
-                from .utils import label_smoothed_nll_loss
-            except ImportError:
-                from utils import label_smoothed_nll_loss
+            from utils import label_smoothed_nll_loss

            self.loss_fn = label_smoothed_nll_loss