From 5c7c044dd99f85fcaa63ca7a862b6604af0d0376 Mon Sep 17 00:00:00 2001
From: Hieu Hoang <hieuhoang@gmail.com>
Date: Wed, 14 Sep 2016 08:36:31 +0200
Subject: [PATCH] debug

---
 src/tensor.h | 2 +-
 src/test.cu  | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/tensor.h b/src/tensor.h
index 2cf4c267..4c6b89dd 100644
--- a/src/tensor.h
+++ b/src/tensor.h
@@ -155,7 +155,7 @@ class TensorImpl {
     void set(const std::vector<Float> &values) {
 	  size_t totSize = std::accumulate(shape().begin(), shape().end(),
 			  1, std::multiplies<int>());
-	  std::cerr << "totSize=" << totSize << " " << values.size() << std::endl;
+	  std::cerr << "tensor size=" << totSize << " vector size=" << values.size() << std::endl;
 	  assert(totSize == values.size());
 	  thrust::copy(values.begin(), values.end(), data_.begin());
     }
diff --git a/src/test.cu b/src/test.cu
index d87c9d1a..dd65d256 100644
--- a/src/test.cu
+++ b/src/test.cu
@@ -31,6 +31,7 @@ int main(int argc, char** argv) {
   int numImg, imgSize;
   vector<float> images = datasets::mnist::ReadImages("../examples/mnist/t10k-images-idx3-ubyte", numImg, imgSize);
   vector<int> labels = datasets::mnist::ReadLabels("../examples/mnist/t10k-labels-idx1-ubyte");
+  cerr << "images=" << images.size() << " labels=" << labels.size() << endl;
   tx.Load(images);
   //ty.Load(labels);