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

fix typo in readme

Browse Source
This commit is contained in:
thomwolf
2019-02-11 13:37:12 +01:00
parent e8fe6b7140
commit 81c7e3ec9f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -350,7 +350,7 @@ with torch.no_grad():
# get the predicted last token # get the predicted last token
predicted_index = torch.argmax(predictions_2[0, -1, :]).item() predicted_index = torch.argmax(predictions_2[0, -1, :]).item()
predicted_token = tokenizer.convert_ids_to_tokens([predicted_index])[0] predicted_token = tokenizer.convert_ids_to_tokens([predicted_index])[0]
assert predicted_token == '.</w>' assert predicted_token == 'who'
``` ```
## Doc ## Doc