Use SGD to avoid OOM on CI

dmlc · Aug 27, 2018 · 968591a · 968591a
1 parent 87e8dd6
commit 968591a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/tests/test_scripts.py b/scripts/tests/test_scripts.py
@@ -30,7 +30,7 @@ def test_toy():
 @pytest.mark.serial
 def test_embedding():
     process = subprocess.check_call(['python', './scripts/word_embeddings/train_fasttext.py',
-                                     '--gpu', '0', '--epochs', '1'])
+                                     '--gpu', '0', '--epochs', '1', '--optimizer', 'sgd'])
 
 
 @pytest.mark.serial