ml.shifu.guagua.yarn.example.nn

Class NNMaster

java.lang.Object

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

All Implemented Interfaces:

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.

Constructor Summary

Constructors

Constructor and Description
NNMaster()

Method Summary

Methods

Modifier and Type	Method and Description
NNParams	compute(ml.shifu.guagua.master.MasterContext<NNParams,NNParams> context)

Methods inherited from class java.lang.Object

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

Constructor Detail

NNMaster

public NNMaster()

Method Detail

compute

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

Specified by:

compute in interface ml.shifu.guagua.master.MasterComputable<NNParams,NNParams>