Commit 7b8abfa1 authored by Wael Ramadan's avatar Wael Ramadan
Browse files

use_gpu false by default

parent 73640c72
Pipeline #6893 passed with stage
in 5 minutes and 32 seconds
...@@ -25,7 +25,8 @@ from texta_mlp.settings import ( ...@@ -25,7 +25,8 @@ from texta_mlp.settings import (
SUPPORTED_ANALYZERS, SUPPORTED_ANALYZERS,
DEFAULT_ANALYZERS, DEFAULT_ANALYZERS,
ENTITY_MAPPER_DATA_URLS, ENTITY_MAPPER_DATA_URLS,
CONCATENATOR_DATA_FILES CONCATENATOR_DATA_FILES,
USE_GPU
) )
...@@ -39,7 +40,7 @@ class MLP: ...@@ -39,7 +40,7 @@ class MLP:
resource_dir: str = DEFAULT_RESOURCE_DIR, resource_dir: str = DEFAULT_RESOURCE_DIR,
ner_model_langs: list = CUSTOM_NER_MODEL_LANGS, ner_model_langs: list = CUSTOM_NER_MODEL_LANGS,
logging_level: str ="error", logging_level: str ="error",
use_gpu: bool = True, use_gpu: bool = USE_GPU,
gpu_device_id: int = 0, gpu_device_id: int = 0,
refresh_data: bool = REFRESH_DATA refresh_data: bool = REFRESH_DATA
): ):
......
...@@ -66,3 +66,6 @@ STANZA_NER_SUPPORT = ("ar", "zh", "nl", "en", "fr", "de", "ru", "es", "uk") ...@@ -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. # Here we add langs that will have custom ner models.
CUSTOM_NER_MODEL_LANGS = ["et"] CUSTOM_NER_MODEL_LANGS = ["et"]
# Use gpu for pytorch
USE_GPU = parse_bool_env("TEXTA_MLP_USE_GPU", False)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment