explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-09-22 03:49:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

Correct parser.py use_upper param info (#9180)

Browse Source
This commit is contained in:
j-frei 2021-09-10 16:19:58 +02:00 committed by svlandeg
parent 9c4e84d4a1
commit 5d0cc0d2ab
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.