minor update
authorYangqing Jia <jiayq84@gmail.com>
Thu, 19 Sep 2013 21:55:58 +0000 (14:55 -0700)
committerYangqing Jia <jiayq84@gmail.com>
Thu, 19 Sep 2013 21:55:58 +0000 (14:55 -0700)
src/caffeine/test/test_neuron_layer.cpp

index 5a28b0056cc24e4c10300089f8fd152b19ceb366..b4254fec4070083064fef2f51791772b1951bc33 100644 (file)
@@ -158,7 +158,9 @@ TYPED_TEST(NeuronLayerTest, TestDropoutGradientGPU) {
     Caffeine::set_mode(Caffeine::GPU);
     DropoutLayer<TypeParam> layer(layer_param);
     GradientChecker<TypeParam> checker(1e-2, 1e-3);
-    checker.CheckGradientExhaustive(layer, this->blob_bottom_vec_, this->blob_top_vec_);
+    // it is too expensive to call curand multiple times, so we don't do an
+    // exhaustive gradient check.
+    checker.CheckGradient(layer, this->blob_bottom_vec_, this->blob_top_vec_);
   } else {
     LOG(ERROR) << "Skipping test to spare my laptop.";
   }