Merge pull request #1953 from zoq/AtrousConvolutionTestFix

AtrousConvolutionLayer stablize test.
This commit is contained in:
Marcus Edel
2019-08-01 20:43:18 +02:00
committed by GitHub
2 changed files with 14 additions and 5 deletions
+13 -4
View File
@@ -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
+1 -1
View File
@@ -1921,7 +1921,7 @@ BOOST_AUTO_TEST_CASE(GradientAtrousConvolutionLayerTest)
model = new FFN<NegativeLogLikelihood<>, RandomInitialization>();
model->Predictors() = input;
model->Responses() = target;
model->Add<Linear<> >(36, 36);
model->Add<IdentityLayer<> >();
model->Add<AtrousConvolution<> >(1, 1, 3, 3, 1, 1, 0, 0, 6, 6, 2, 2);
model->Add<LogSoftMax<> >();
}