Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Correct parser.py use_upper param info (#9180)

This commit is contained in:
j-frei 2021-09-10 16:19:58 +02:00 committed by GitHub
parent aba6ce3a43
commit 462b009648
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/ml/models/parser.py
View File

@ -56,7 +56,7 @@ def build_tb_parser_model(
non-linearity if use_upper=False. non-linearity if use_upper=False.
use_upper (bool): Whether to use an additional hidden layer after the state use_upper (bool): Whether to use an additional hidden layer after the state
vector in order to predict the action scores. It is recommended to set vector in order to predict the action scores. It is recommended to set
this to False for large pretrained models such as transformers, and False this to False for large pretrained models such as transformers, and True
for smaller networks. The upper layer is computed on CPU, which becomes for smaller networks. The upper layer is computed on CPU, which becomes
a bottleneck on larger GPU-based models, where it's also less necessary. a bottleneck on larger GPU-based models, where it's also less necessary.
nO (int or None): The number of actions the model will predict between. nO (int or None): The number of actions the model will predict between.