Fix test for changed cudnn option.

cmusatyalab · Jun 7, 2016 · 288fd8a · 288fd8a
1 parent bced2bb
commit 288fd8a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/openface_neural_net_training_tests.py b/tests/openface_neural_net_training_tests.py
@@ -64,7 +64,7 @@ def test_dnn_training():
            '-nEpochs', '10',
            '-epochSize', '1',
            '-cache', netWorkDir,
-           '-cuda', '-cudnn', '-testing',
+           '-cuda', '-testing',
            '-nDonkeys', '-1']
     p = Popen(cmd, stdout=PIPE, stderr=PIPE,
               cwd=os.path.join(openfaceDir, 'training'))