SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Patrick von Platen
2020-11-03 13:56:41 +01:00
committed by GitHub
parent 504ff7bb12
commit 9f1747f999
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/seq2seq/seq2seq_trainer.py
View File

@@ -62,10 +62,7 @@ class Seq2SeqTrainer(Trainer):
self.loss_fn = torch.nn.CrossEntropyLoss(ignore_index=self.config.pad_token_id) self.loss_fn = torch.nn.CrossEntropyLoss(ignore_index=self.config.pad_token_id)
else: else:
# dynamically import label_smoothed_nll_loss # dynamically import label_smoothed_nll_loss
try: from utils import label_smoothed_nll_loss
from .utils import label_smoothed_nll_loss
except ImportError:
from utils import label_smoothed_nll_loss
self.loss_fn = label_smoothed_nll_loss self.loss_fn = label_smoothed_nll_loss