kd13
/

RoPERT-MLM-mini

Model card Files Files and versions

kd13 commited on 19 days ago

Commit

95ffa1d

·

verified ·

1 Parent(s): a70e122

Update modeling_mybert.py

Files changed (1) hide show

modeling_mybert.py +1 -1

modeling_mybert.py CHANGED Viewed

@@ -272,4 +272,4 @@ class MyBertForMaskedLM(MyBertPreTrainedModel):
             output = (prediction_scores,)
             return ((loss,) + output) if loss is not None else output
-        return MaskedLMOutput(loss=loss, logits=prediction_scores)

             output = (prediction_scores,)
             return ((loss,) + output) if loss is not None else output
+        return MaskedLMOutput(loss=loss, logits=prediction_scores)