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

Just some typos

Browse Source
This commit is contained in:
Gabriel Luiz Freitas Almeida
2019-09-27 07:08:36 -03:00
committed by GitHub
parent d83d295763
commit d2de5b9d8c
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
docs/source/quickstart.md
View File

@@ -93,8 +93,8 @@ Let's see how we can use `BertModel` to encode our inputs in hidden-states:
# Load pre-trained model (weights) # Load pre-trained model (weights)
model = BertModel.from_pretrained('bert-base-uncased') model = BertModel.from_pretrained('bert-base-uncased')
# Set the model in evaluation mode to desactivate the DropOut modules # Set the model in evaluation mode to deactivate the DropOut modules
# This is IMPORTANT to have reproductible results during evaluation! # This is IMPORTANT to have reproducible results during evaluation!
model.eval() model.eval()
# If you have a GPU, put everything on cuda # If you have a GPU, put everything on cuda
@@ -168,8 +168,8 @@ Let's see how to use `GPT2LMHeadModel` to generate the next token following our
# Load pre-trained model (weights) # Load pre-trained model (weights)
model = GPT2LMHeadModel.from_pretrained('gpt2') model = GPT2LMHeadModel.from_pretrained('gpt2')
# Set the model in evaluation mode to desactivate the DropOut modules # Set the model in evaluation mode to deactivate the DropOut modules
# This is IMPORTANT to have reproductible results during evaluation! # This is IMPORTANT to have reproducible results during evaluation!
model.eval() model.eval()
# If you have a GPU, put everything on cuda # If you have a GPU, put everything on cuda