fix typo in readme

This commit is contained in:
thomwolf
2019-02-11 13:37:12 +01:00
parent e8fe6b7140
commit 81c7e3ec9f

View File

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