Skip to content
GitLab
Menu
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
texta
texta-mlp-python
Commits
7b8abfa1
Commit
7b8abfa1
authored
Oct 23, 2021
by
Wael Ramadan
Browse files
use_gpu false by default
parent
73640c72
Pipeline
#6893
passed with stage
in 5 minutes and 32 seconds
Changes
2
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
texta_mlp/mlp.py
View file @
7b8abfa1
...
...
@@ -25,7 +25,8 @@ from texta_mlp.settings import (
SUPPORTED_ANALYZERS
,
DEFAULT_ANALYZERS
,
ENTITY_MAPPER_DATA_URLS
,
CONCATENATOR_DATA_FILES
CONCATENATOR_DATA_FILES
,
USE_GPU
)
...
...
@@ -39,7 +40,7 @@ class MLP:
resource_dir
:
str
=
DEFAULT_RESOURCE_DIR
,
ner_model_langs
:
list
=
CUSTOM_NER_MODEL_LANGS
,
logging_level
:
str
=
"error"
,
use_gpu
:
bool
=
True
,
use_gpu
:
bool
=
USE_GPU
,
gpu_device_id
:
int
=
0
,
refresh_data
:
bool
=
REFRESH_DATA
):
...
...
texta_mlp/settings.py
View file @
7b8abfa1
...
...
@@ -66,3 +66,6 @@ STANZA_NER_SUPPORT = ("ar", "zh", "nl", "en", "fr", "de", "ru", "es", "uk")
# Here we add langs that will have custom ner models.
CUSTOM_NER_MODEL_LANGS
=
[
"et"
]
# Use gpu for pytorch
USE_GPU
=
parse_bool_env
(
"TEXTA_MLP_USE_GPU"
,
False
)
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment