From 1c9f6f7f9902ce05e71fd62edea9b10f895009e5 Mon Sep 17 00:00:00 2001 From: liaogang Date: Wed, 9 Nov 2016 20:36:45 +0800 Subject: [PATCH] Update --- paddle/gserver/tests/test_LayerGrad.cpp | 47 +++++++++++++------------ 1 file changed, 25 insertions(+), 22 deletions(-) diff --git a/paddle/gserver/tests/test_LayerGrad.cpp b/paddle/gserver/tests/test_LayerGrad.cpp index d8c45040d8..ad09d7d00e 100644 --- a/paddle/gserver/tests/test_LayerGrad.cpp +++ b/paddle/gserver/tests/test_LayerGrad.cpp @@ -32,28 +32,6 @@ P_DECLARE_double(checkgrad_eps); P_DECLARE_bool(thread_local_rand_use_global_seed); P_DECLARE_bool(prev_batch_state); -TEST(Layer, BilinearInterpLayer) { - TestConfig config; - config.layerConfig.set_type("bilinear_interp"); - config.biasSize = 0; - - config.inputDefs.push_back({INPUT_DATA, "layer_0", 4096, 0}); - LayerInputConfig* input = config.layerConfig.add_inputs(); - BilinearInterpConfig* bilinear = input->mutable_bilinear_interp_conf(); - - bilinear->set_img_size_x(32); - bilinear->set_img_size_y(32); - bilinear->set_num_channels(4); - - for (auto useGpu : {false, true}) { - for (auto out_size : {32, 64, 128}) { - bilinear->set_out_size_x(out_size); - bilinear->set_out_size_y(out_size); - testLayerGrad(config, "bilinear_interp", 10, false, useGpu); - } - } -} - TEST(Operator, dot_mul) { TestConfig config; config.layerConfig.set_size(10); @@ -197,6 +175,31 @@ TEST(Projection, conv) { } #endif +TEST(Layer, BilinearInterpLayer) { + TestConfig config; + config.layerConfig.set_type("bilinear_interp"); + config.biasSize = 0; + + config.inputDefs.push_back({INPUT_DATA, "layer_0", 4096, 0}); + + for (auto useGpu : {false, true}) { + for (auto out_size : {32, 64}) { + LOG(INFO) << " out_size_x=" << out_size + << " out_size_y=" << out_size; + LayerInputConfig* input + = config.layerConfig.add_inputs(); + BilinearInterpConfig* bilinear + = input->mutable_bilinear_interp_conf(); + bilinear->set_img_size_x(32); + bilinear->set_img_size_y(32); + bilinear->set_num_channels(4); + bilinear->set_out_size_x(out_size); + bilinear->set_out_size_y(out_size); + testLayerGrad(config, "bilinear_interp", 10, false, useGpu); + } + } +} + TEST(Layer, concat) { TestConfig config; config.biasSize = 0; -- GitLab