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

Change how to load config of XLNetLMHeadModel (#14746)

Browse Source
This commit is contained in:
Josué Nascimento
2021-12-13 14:34:26 -03:00
committed by GitHub
parent 15a9d01519
commit 971e36667a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pytorch/language-modeling/run_plm.py
View File

@@ -348,7 +348,7 @@ def main():
) )
else: else:
logger.info("Training new model from scratch") logger.info("Training new model from scratch")
model = XLNetLMHeadModel.from_config(config) model = XLNetLMHeadModel(config)
model.resize_token_embeddings(len(tokenizer)) model.resize_token_embeddings(len(tokenizer))