From ca0914ce975aa4a906cc91d88ea15ec4626bcc6c Mon Sep 17 00:00:00 2001
From: Joe Zhao <ztuowen@gmail.com>
Date: Mon, 6 Oct 2014 19:41:36 +0800
Subject: Virtual 8
---
src/CuTNetLib/cuCompDisc.cc | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
(limited to 'src')
diff --git a/src/CuTNetLib/cuCompDisc.cc b/src/CuTNetLib/cuCompDisc.cc
index 4c31f58..2c0dbe3 100644
--- a/src/CuTNetLib/cuCompDisc.cc
+++ b/src/CuTNetLib/cuCompDisc.cc
@@ -1,5 +1,5 @@
-
+#include <stdio.h>
#include "cuCompDisc.h"
#include "cuNetwork.h"
@@ -13,6 +13,7 @@ namespace TNet
CuDiscrete::
Propagate()
{
+ printf("Propagate\n");
for (int i=0;i<inID.size(); i++)
mBlocks[inID[i].block]->SetInput(GetInput(i),inID[i].pos);
for (int i=0; i<mBlocks.size(); i++)
@@ -30,6 +31,7 @@ namespace TNet
CuDiscrete::
Backpropagate()
{
+ printf("BackPropagate\n");
for (int i=0;i<outID.size(); i++)
mBlocks[outID[i].block]->SetErrorInput(GetOutput(i),outID[i].pos);
for(int i=0; i<mBlocks.size(); i++)
@@ -47,6 +49,7 @@ namespace TNet
CuDiscrete::
Update()
{
+ printf("update\n");
for(int i=0; i<mBlocks.size(); i++)
if ( mBlocks[i]->IsUpdatable() )
{
--
cgit v1.2.3-70-g09d2