diff options
Diffstat (limited to 'src/CuTNetLib/cuCompDisc.cc')
| -rw-r--r-- | src/CuTNetLib/cuCompDisc.cc | 2 |
1 files changed, 2 insertions, 0 deletions
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 |