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

update doc

Browse Source
This commit is contained in:
VictorSanh
2019-05-31 01:04:16 -04:00
parent 98f5c7864f
commit 45d21502f0
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
hubconfs/gpt_hubconf.py
View File

@@ -146,6 +146,7 @@ def openAIGPTLMHeadModel(*args, **kwargs):
# Get the predicted last token # Get the predicted last token
>>> predicted_index = torch.argmax(predictions[0, -1, :]).item() >>> predicted_index = torch.argmax(predictions[0, -1, :]).item()
>>> predicted_token = tokenizer.convert_ids_to_tokens([predicted_index])[0] >>> predicted_token = tokenizer.convert_ids_to_tokens([predicted_index])[0]
'.</w>'
""" """
model = OpenAIGPTLMHeadModel.from_pretrained(*args, **kwargs) model = OpenAIGPTLMHeadModel.from_pretrained(*args, **kwargs)
return model return model