Reduce number of ngram buckets to 1000 due to OOM

dmlc · Aug 28, 2018 · dc486b4 · dc486b4
1 parent 0ff68c5
commit dc486b4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/scripts/tests/test_scripts.py b/scripts/tests/test_scripts.py
@@ -29,8 +29,10 @@ def test_toy():
 
 @pytest.mark.serial
 def test_embedding():
-    process = subprocess.check_call(['python', './scripts/word_embeddings/train_fasttext.py',
-                                     '--gpu', '0', '--epochs', '1', '--optimizer', 'sgd'])
+    process = subprocess.check_call([
+        'python', './scripts/word_embeddings/train_fasttext.py', '--gpu', '0',
+        '--epochs', '1', '--optimizer', 'sgd', '--ngram-buckets', '1000'
+    ])
 
 
 @pytest.mark.serial