jingtaozhan / DRhard

SIGIR'21: Optimizing DR with hard negatives and achieving SOTA first-stage retrieval performance on TREC DL Track.
BSD 3-Clause "New" or "Revised" License
125 stars 14 forks source link

model = RobertaDot.from_pretrained(model_path, config=config).cuda() #2

Closed wangjiajia5889758 closed 3 years ago

wangjiajia5889758 commented 3 years ago

when I load the model file, I get the following problem which I don't understand: Some weights of the model checkpoint at /home/tu/device/backup/wjjia/DRhard/adore-star were not used when initializing RobertaDot: ['classifier.dense.weight', 'classifier.dense.bias', 'classifier.out_proj.weight', 'classifier.out_proj.bias', 'roberta.pooler.dense.weight', 'roberta.pooler.dense.bias']

jingtaozhan commented 3 years ago

This is expected and you can just ignore it.