public override void Init(Int32 nGPU)
{
m_trainingStep = 0;
if (Owner.LearningMethod == MyAbstractFeedForwardNode.MyLearningMethod.GRADIENT_DESCENT)
{
MyGradientBackPropAgent agent = new MyGradientBackPropAgent(Owner, nGPU, Owner.TrainingLabel);
BackPropAgent = agent;
agent.LearningBatchSize = LearningBatchSize;
agent.LearningDuration = LearningDuration;
agent.LearningRate = LearningRate;
agent.LearningMomentum = LearningMomentum;
}
else
{
throw new NotImplementedException("Unknown learning method.");
}
}
