From: Yangqing Jia <jiayq84@gmail.com>
Date: Thu, 19 Sep 2013 21:55:58 +0000 (-0700)
Subject: minor update
X-Git-Url: https://git.ti.com/gitweb?p=jacinto-ai%2Fcaffe-jacinto.git;a=commitdiff_plain;h=cef51df08a3918f4f91c501ee4bee4e288cebfb4

minor update
---

diff --git a/src/caffeine/test/test_neuron_layer.cpp b/src/caffeine/test/test_neuron_layer.cpp
index 5a28b005..b4254fec 100644
--- a/src/caffeine/test/test_neuron_layer.cpp
+++ b/src/caffeine/test/test_neuron_layer.cpp
@@ -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.";
   }