VarunGumma commited on
Commit
c0b8106
1 Parent(s): 0132475

Update modeling_indictrans.py

Browse files
Files changed (1) hide show
  1. modeling_indictrans.py +1 -1
modeling_indictrans.py CHANGED
@@ -1393,7 +1393,7 @@ class IndicTransForConditionalGeneration(IndicTransPreTrainedModel):
1393
  masked_lm_loss = F.cross_entropy(
1394
  input=lm_logits.view(-1, self.config.decoder_vocab_size),
1395
  target=labels.view(-1),
1396
- ignore_index=self.config.pad_token_id,
1397
  label_smoothing=self._label_smoothing,
1398
  )
1399
 
 
1393
  masked_lm_loss = F.cross_entropy(
1394
  input=lm_logits.view(-1, self.config.decoder_vocab_size),
1395
  target=labels.view(-1),
1396
+ ignore_index=-100,
1397
  label_smoothing=self._label_smoothing,
1398
  )
1399