From 462b009648d9dfaadb5ac0fe77f59c69e56f2c2f Mon Sep 17 00:00:00 2001 From: j-frei <27365662+j-frei@users.noreply.github.com> Date: Fri, 10 Sep 2021 16:19:58 +0200 Subject: [PATCH] Correct parser.py use_upper param info (#9180) --- spacy/ml/models/parser.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spacy/ml/models/parser.py b/spacy/ml/models/parser.py index 80751a695..97137313d 100644 --- a/spacy/ml/models/parser.py +++ b/spacy/ml/models/parser.py @@ -56,7 +56,7 @@ def build_tb_parser_model( non-linearity if use_upper=False. 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 - 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 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.