org.nd4j.parameterserver.distributed.training

Class BaseTrainer<T extends TrainingMessage>

java.lang.Object

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

All Implemented Interfaces:

TrainingDriver<T>

Direct Known Subclasses:

CbowTrainer, SkipGramTrainer

public abstract class BaseTrainer<T extends TrainingMessage>
extends Object
implements TrainingDriver<T>

Author:

raver119@gmail.co,

Field Summary

Fields

Modifier and Type	Field and Description
protected Clipboard	clipboard
protected FrameCompletionHandler	completionHandler
protected Storage	storage
protected Transport	transport
protected VoidConfiguration	voidConfiguration

Constructor Summary

Constructors

Constructor and Description
BaseTrainer()

Method Summary

Modifier and Type	Method and Description
void	addCompletionHook(long originatorId, long frameId, long messageId)
void	init(VoidConfiguration voidConfiguration, Transport transport, Storage storage, Clipboard clipboard)
protected int[]	replicate(int value, int size)

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.nd4j.parameterserver.distributed.training.TrainingDriver

aggregationFinished, finishTraining, pickTraining, startTraining, targetMessageClass

Field Detail

voidConfiguration

protected VoidConfiguration voidConfiguration

transport

protected Transport transport

clipboard

protected Clipboard clipboard

storage

protected Storage storage

completionHandler

protected FrameCompletionHandler completionHandler

Constructor Detail

BaseTrainer

public BaseTrainer()

Method Detail

init

public void init(@NonNull
                 VoidConfiguration voidConfiguration,
                 @NonNull
                 Transport transport,
                 @NonNull
                 Storage storage,
                 @NonNull
                 Clipboard clipboard)

Specified by:

init in interface TrainingDriver<T extends TrainingMessage>

replicate

protected int[] replicate(int value,
                          int size)

addCompletionHook

public void addCompletionHook(long originatorId,
                              long frameId,
                              long messageId)

Specified by:

addCompletionHook in interface TrainingDriver<T extends TrainingMessage>