Package org.deeplearning4j.parallelism.parameterserver

Class ParameterServerTrainer

java.lang.Object

java.lang.Thread

org.deeplearning4j.parallelism.trainer.DefaultTrainer

org.deeplearning4j.parallelism.parameterserver.ParameterServerTrainer

All Implemented Interfaces:

Runnable, Trainer

public class ParameterServerTrainer
extends DefaultTrainer

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	ParameterServerTrainer.ParameterServerTrainerBuilder

Nested classes/interfaces inherited from class org.deeplearning4j.parallelism.trainer.DefaultTrainer

DefaultTrainer.DefaultTrainerBuilder

Nested classes/interfaces inherited from class java.lang.Thread

Thread.State, Thread.UncaughtExceptionHandler

Field Summary

Fields inherited from class org.deeplearning4j.parallelism.trainer.DefaultTrainer

averagingFrequency, isStopped, lastEtlTime, modelLock, nullDataSet, nullMode, onRootModel, originalModel, parallelWrapper, queue, queueMDS, replicatedModel, running, shouldStop, shouldUpdate, threadId, thrownException, useMDS, uuid, workspaceMode

Fields inherited from class java.lang.Thread

MAX_PRIORITY, MIN_PRIORITY, NORM_PRIORITY

Constructor Summary

Constructors

Constructor	Description
ParameterServerTrainer()

Method Summary

Modifier and Type	Method	Description
void	feedDataSet(@NonNull org.nd4j.linalg.dataset.api.DataSet dataSet, long time)
void	feedMultiDataSet(@NonNull org.nd4j.linalg.dataset.api.MultiDataSet dataSet, long time)
org.deeplearning4j.nn.api.Model	getModel()
void	updateModel(@NonNull org.deeplearning4j.nn.api.Model model)

Methods inherited from class org.deeplearning4j.parallelism.trainer.DefaultTrainer

averagingRequired, builder, cloneListener, configureListeners, fit, fit, getUuid, isRunning, postInit, run, setupIfNeccessary, shutdown, updateModelParams, updateUpdaterParams, waitTillRunning

Methods inherited from class java.lang.Thread

activeCount, checkAccess, clone, countStackFrames, currentThread, dumpStack, enumerate, getAllStackTraces, getContextClassLoader, getDefaultUncaughtExceptionHandler, getId, getName, getPriority, getStackTrace, getState, getThreadGroup, getUncaughtExceptionHandler, holdsLock, interrupt, interrupted, isAlive, isDaemon, isInterrupted, join, join, join, onSpinWait, resume, setContextClassLoader, setDaemon, setDefaultUncaughtExceptionHandler, setName, setPriority, setUncaughtExceptionHandler, sleep, sleep, start, stop, suspend, toString, yield

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.deeplearning4j.parallelism.trainer.Trainer

setUncaughtExceptionHandler, start

Constructor Detail

ParameterServerTrainer

public ParameterServerTrainer()

Method Detail

feedMultiDataSet

public void feedMultiDataSet(@NonNull
                             @NonNull org.nd4j.linalg.dataset.api.MultiDataSet dataSet,
                             long time)

Specified by:

feedMultiDataSet in interface Trainer

Overrides:

feedMultiDataSet in class DefaultTrainer

feedDataSet

public void feedDataSet(@NonNull
                        @NonNull org.nd4j.linalg.dataset.api.DataSet dataSet,
                        long time)

Specified by:

feedDataSet in interface Trainer

Overrides:

feedDataSet in class DefaultTrainer

getModel

public org.deeplearning4j.nn.api.Model getModel()

Specified by:

getModel in interface Trainer

Overrides:

getModel in class DefaultTrainer

updateModel

public void updateModel(@NonNull
                        @NonNull org.deeplearning4j.nn.api.Model model)

Specified by:

updateModel in interface Trainer

Overrides:

updateModel in class DefaultTrainer