diff --git a/src/mlpack/tests/CMakeLists.txt b/src/mlpack/tests/CMakeLists.txt index 2ffe0e641d..80c1c231e1 100644 --- a/src/mlpack/tests/CMakeLists.txt +++ b/src/mlpack/tests/CMakeLists.txt @@ -187,11 +187,20 @@ add_custom_command(TARGET mlpack_test # The list of long running parallel tests set(parallel_tests + "ANNLayerTest;" "AsyncLearningTest;" - "SVDIncrementalTest;SVDBatchTest;" - "LocalCoordinateCodingTest;FeedForwardNetworkTest;SparseAutoencoderTest;" - "GMMTest;CFTest;ConvolutionalNetworkTest;HMMTest;LARSTest;" - "LogisticRegressionTest;GmmTrainMainTest;" + "SVDIncrementalTest;" + "SVDBatchTest;" + "LocalCoordinateCodingTest;" + "FeedForwardNetworkTest;" + "SparseAutoencoderTest;" + "GMMTest;" + "CFTest;" + "ConvolutionalNetworkTest;" + "HMMTest;" + "LARSTest;" + "LogisticRegressionTest;" + "GmmTrainMainTest;" "LinearSVMTest") # Add tests to the testing framework diff --git a/src/mlpack/tests/ann_layer_test.cpp b/src/mlpack/tests/ann_layer_test.cpp index c936fab3f8..0cc30f63e8 100644 --- a/src/mlpack/tests/ann_layer_test.cpp +++ b/src/mlpack/tests/ann_layer_test.cpp @@ -1921,7 +1921,7 @@ BOOST_AUTO_TEST_CASE(GradientAtrousConvolutionLayerTest) model = new FFN, RandomInitialization>(); model->Predictors() = input; model->Responses() = target; - model->Add >(36, 36); + model->Add >(); model->Add >(1, 1, 3, 3, 1, 1, 0, 0, 6, 6, 2, 2); model->Add >(); }