Merge pull request inspirehep#5 from michamos/no-cuda

config: don't use CUDA by default
salmanmaq · Oct 24, 2018 · 3c9f6b0 · 3c9f6b0
2 parents 661cfa3 + 3882ae9
commit 3c9f6b0
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/inspire_classifier/config.py b/inspire_classifier/config.py
@@ -30,7 +30,7 @@
 CLASSIFIER_CLASSIFIER_CYCLE_LENGTH = 14
 CLASSIFIER_LANGUAGE_MODEL_BATCH_SIZE = 32
 CLASSIFIER_CLASSIFIER_BATCH_SIZE = 10
-CLASSIFIER_CUDA_DEVICE_ID = 0
+CLASSIFIER_CUDA_DEVICE_ID = -1  # set to 0 to use a GPU
 
 CLASSIFIER_DATA_PATH = 'data'
 CLASSIFIER_LANGUAGE_MODEL_PATH = 'models/language_model'

diff --git a/tests/integration/conftest.py b/tests/integration/conftest.py
@@ -43,7 +43,6 @@ def app():
         app.config['CLASSIFIER_CLASSIFIER_CYCLE_LENGTH'] = 1
         app.config['CLASSIFIER_LANGUAGE_MODEL_BATCH_SIZE'] = 10
         app.config['CLASSIFIER_CLASSIFIER_BATCH_SIZE'] = 10
-        app.config['CLASSIFIER_CUDA_DEVICE_ID'] = -1
         app.config['CLASSIFIER_VALIDATION_DATA_FRACTION'] = 0.2
         yield app