org.nd4j.parameterserver.distributed.training.impl

Class CbowTrainer

java.lang.Object

org.nd4j.parameterserver.distributed.training.BaseTrainer<CbowRequestMessage>

org.nd4j.parameterserver.distributed.training.impl.CbowTrainer

All Implemented Interfaces:

TrainingDriver<CbowRequestMessage>

public class CbowTrainer
extends BaseTrainer<CbowRequestMessage>

Author:

raver119@gmail.com

Field Summary

Fields

Modifier and Type	Field and Description
protected Map<RequestDescriptor,CbowChain>	chains
protected AtomicLong	cntRounds

Fields inherited from class org.nd4j.parameterserver.distributed.training.BaseTrainer

clipboard, completionHandler, storage, transport, voidConfiguration

Constructor Summary

Constructors

Constructor and Description
CbowTrainer()

Method Summary

Modifier and Type	Method and Description
void	aggregationFinished(VoidAggregation aggregation)
void	finishTraining(long originatorId, long taskId)
void	pickTraining(CbowRequestMessage message)
void	startTraining(CbowRequestMessage message)
String	targetMessageClass()

Methods inherited from class org.nd4j.parameterserver.distributed.training.BaseTrainer

addCompletionHook, init, replicate

Methods inherited from class java.lang.Object

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

Field Detail

chains

protected Map<RequestDescriptor,CbowChain> chains

cntRounds

protected AtomicLong cntRounds

Constructor Detail

CbowTrainer

public CbowTrainer()

Method Detail

startTraining

public void startTraining(CbowRequestMessage message)

pickTraining

public void pickTraining(CbowRequestMessage message)

aggregationFinished

public void aggregationFinished(VoidAggregation aggregation)

finishTraining

public void finishTraining(long originatorId,
                           long taskId)

targetMessageClass

public String targetMessageClass()