Merge pull request #1953 from zoq/AtrousConvolutionTestFix
AtrousConvolutionLayer stablize test.
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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<> >();
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user