From 04a620db191e933dd8edb78e3386aefa4225a9c0 Mon Sep 17 00:00:00 2001
From: Hieu Hoang <s0565741@odin.inf.ed.ac.uk>
Date: Wed, 14 Sep 2016 13:42:59 +0100
Subject: [PATCH] debug

---
 src/test.cu | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/test.cu b/src/test.cu
index dec93919..bd417cee 100644
--- a/src/test.cu
+++ b/src/test.cu
@@ -24,7 +24,7 @@ int main(int argc, char** argv) {
   Expr scores = dot(x, w) + b;
   Expr lr = softmax(scores, axis=1, name="pred");
   Expr graph = -mean(sum(y * log(lr), axis=1), axis=0, name="cost");
-  cerr << "lr=" << Debug(lr.val().shape()) << endl;
+  //cerr << "lr=" << Debug(lr.val().shape()) << endl;
 
   int numofdata;
   vector<float> images = datasets::mnist::ReadImages("../examples/mnist/t10k-images-idx3-ubyte", numofdata, IMAGE_SIZE);
@@ -50,7 +50,8 @@ int main(int argc, char** argv) {
   std::cerr << "lr: " << Debug(lr.val().shape()) << endl;
   std::cerr << "Log-likelihood: " << Debug(graph.val().shape()) << endl ;
 
-  std::cerr << "scores=" << scores.val().Debug() << endl;
+  //std::cerr << "scores=" << scores.val().Debug() << endl;
+  std::cerr << "lr=" << lr.val().Debug() << endl;
 
   graph.backward();