ml.shifu.guagua.yarn.example.nn
类 NNMaster

java.lang.Object
  ml.shifu.guagua.yarn.example.nn.NNMaster

所有已实现的接口：

ml.shifu.guagua.master.MasterComputable<NNParams,NNParams>

public class NNMaster
extends Object
implements ml.shifu.guagua.master.MasterComputable<NNParams,NNParams>

NNMaster is used to accumulate all workers NN parameters.

All gradients are accumulated from workers to calculate model weights, and then new weights are sent to workers. Then workers use new weights to set their models and train for another iteration.

This logic follows Encog multi-core implementation.

To make sure workers and master use the same initialization weights, first iteration of this guagua application is used to compute initial weights which are then sent to works for their initial weights.

构造方法摘要

NNMaster()

方法摘要

NNParams

compute(ml.shifu.guagua.master.MasterContext<NNParams,NNParams> context)

从类 java.lang.Object 继承的方法

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

构造方法详细信息

NNMaster

public NNMaster()

方法详细信息

compute

public NNParams compute(ml.shifu.guagua.master.MasterContext<NNParams,NNParams> context)

指定者：

接口 ml.shifu.guagua.master.MasterComputable<NNParams,NNParams> 中的 compute