From 84c51bec05a64b4c3b7be2f68d80d23e6804ed07 Mon Sep 17 00:00:00 2001
From: Joe Zhao <ztuowen@gmail.com>
Date: Mon, 6 Oct 2014 22:39:05 +0800
Subject: Virtual 9+++++++++++++++

---
 src/CuTNetLib/cuCompDisc.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/CuTNetLib/cuCompDisc.cc b/src/CuTNetLib/cuCompDisc.cc
index 55c34a9..9eb08a9 100644
--- a/src/CuTNetLib/cuCompDisc.cc
+++ b/src/CuTNetLib/cuCompDisc.cc
@@ -17,6 +17,7 @@ namespace TNet
       mBlocks[inID[i].block]->SetInput(GetInput(i),inID[i].pos);
     for (int i=0; i<mBlocks.size(); i++)
       mBlocks[i]->Propagate();
+    mErrorOutput.Init(GetOutput());
   }
 
   void 
@@ -34,6 +35,7 @@ namespace TNet
       mBlocks[outID[i].block]->SetErrorInput(GetOutput(i),outID[i].pos);
     for(int i=0; i<mBlocks.size(); i++)
       mBlocks[i]->Backpropagate();
+    mOutput.Init(GetOutput());
   }
   
   void 
-- 
cgit v1.2.3-70-g09d2