fix

diff --git a/src/Makefile b/src/Makefile
index fd8a8ab..3633c71 100644 (file)
--- a/src/Makefile
+++ b/src/Makefile
@@ -46,7 +46,7 @@ NVCC = nvcc ${CXXFLAGS:-fPIC=-Xcompiler -fPIC} $(CPPFLAGS) $(CUDA_ARCH)
 
 all: $(NAME)
 
-linecount:
+linecount: clean
        cloc --read-lang-def=caffeine.cloc caffeine/
 
 test: $(TEST_NAME)

diff --git a/src/caffeine/test/test_gradient_check_util.cpp b/src/caffeine/test/test_gradient_check_util.cpp
index f0a186a..aa540f9 100644 (file)
--- a/src/caffeine/test/test_gradient_check_util.cpp
+++ b/src/caffeine/test/test_gradient_check_util.cpp
@@ -31,7 +31,7 @@ void GradientChecker<Dtype>::CheckGradient(Layer<Dtype>& layer,
   for (int blobid = 0; blobid < blobs_to_check.size(); ++blobid) {
     Blob<Dtype>* current_blob = blobs_to_check[blobid];
     //LOG(ERROR) << "Blob " << blobid << ": checking " << current_blob->count()
-        << " parameters.";
+    //    << " parameters.";
     // go through the values
     for (int feat_id = 0; feat_id < current_blob->count(); ++feat_id) {
       // First, obtain the original data