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

Update quickstart

Browse Source
This commit is contained in:
Lysandre
2020-02-04 11:11:37 -05:00
parent 90ab15cb7a
commit 9c67196b83
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/source/quickstart.md
View File

@@ -299,8 +299,8 @@ model = Model2Model.from_pretrained('fine-tuned-weights')
model.eval() model.eval()
# If you have a GPU, put everything on cuda # If you have a GPU, put everything on cuda
question_tensor = encoded_question.to('cuda') question_tensor = question_tensor.to('cuda')
answer_tensor = encoded_answer.to('cuda') answer_tensor = answer_tensor.to('cuda')
model.to('cuda') model.to('cuda')
# Predict all tokens # Predict all tokens