diff options
| -rw-r--r-- | src/CuTNetLib/cuCompDisc.h | 4 | ||||
| -rw-r--r-- | src/CuTNetLib/cuComponent.h | 4 |
2 files changed, 5 insertions, 3 deletions
diff --git a/src/CuTNetLib/cuCompDisc.h b/src/CuTNetLib/cuCompDisc.h index 937b785..64139c2 100644 --- a/src/CuTNetLib/cuCompDisc.h +++ b/src/CuTNetLib/cuCompDisc.h @@ -40,7 +40,9 @@ namespace TNet { } } - + virtual void Propagate(); + virtual void Backpropagate(); + void Momentum(BaseFloat mmt) { mMomentum = mmt; diff --git a/src/CuTNetLib/cuComponent.h b/src/CuTNetLib/cuComponent.h index fc9666c..9eec461 100644 --- a/src/CuTNetLib/cuComponent.h +++ b/src/CuTNetLib/cuComponent.h @@ -136,10 +136,10 @@ namespace TNet { /// Perform forward pass propagateion Input->Output, /// wrapper for the PropagateFnc method - void Propagate(); + virtual void Propagate(); /// Perform backward pass propagateion ErrorInput->ErrorOutput, /// wrapper for the BackpropagateFnc method - void Backpropagate(); + virtual void Backpropagate(); /// Reads the component parameters from stream virtual void ReadFromStream(std::istream& rIn) { } |