DeBERTa icon indicating copy to clipboard operation
DeBERTa copied to clipboard

[bug] incomplete code

Open shenfe opened this issue 4 years ago • 1 comments

In deberta.mlm, MaskedLayerNorm is not imported from deberta.ops, and PreLayerNorm is undefined.

And I'm not sure if deberta.mlm contains codes for pretraining?

shenfe avatar Jan 14 '21 08:01 shenfe

Same question. I modify the Huggface's code to load all the available values except deberta.embeddings.position_embeddings.weight and run:

from transformers import DebertaTokenizer, DebertaForMaskedLM
import torch

tokenizer = DebertaTokenizer.from_pretrained('microsoft/deberta-base')
model = DebertaForMaskedLM.from_pretrained('microsoft/deberta-base')

inputs = tokenizer("The capital of France is [MASK].", return_tensors="pt")
labels = tokenizer("The capital of France is Paris.", return_tensors="pt")["input_ids"]

outputs = model(**inputs, labels=labels)
print (outputs.loss)

The loss is 3.85, did I do something wrong?

Thanks, Deming

YeDeming avatar Feb 13 '21 09:02 YeDeming