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:

java.lang.Runnable, Trainer

public class ParameterServerTrainer
extends DefaultTrainer

Using an ParameterServerClient we maintain updates for training a neural net. Training happens relative to the mode of the remote ParameterServerNode

Nested Class Summary

Nested Classes

Modifier and Type	Class and 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

java.lang.Thread.State, java.lang.Thread.UncaughtExceptionHandler

Field Summary

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

onRootModel, originalModel, parallelWrapper, queue, queueMDS, replicatedModel, running, shouldStop, shouldUpdate, threadId, thrownException, useMDS, uuid

Fields inherited from class java.lang.Thread

MAX_PRIORITY, MIN_PRIORITY, NORM_PRIORITY

Constructor Summary

Constructors

Constructor and Description
ParameterServerTrainer()

Method Summary

Modifier and Type	Method and Description
void	feedDataSet(org.nd4j.linalg.dataset.api.DataSet dataSet) Train on a DataSet
void	feedMultiDataSet(org.nd4j.linalg.dataset.api.MultiDataSet dataSet) Train on a MultiDataSet
Model	getModel() THe current model for the trainer
void	updateModel(Model model) Update the current Model for the worker

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

cloneListener, configureListeners, isRunning, run, setupIfNeccessary, shutdown, waitTillRunning

Methods inherited from class java.lang.Thread

activeCount, checkAccess, clone, countStackFrames, currentThread, destroy, dumpStack, enumerate, getAllStackTraces, getContextClassLoader, getDefaultUncaughtExceptionHandler, getId, getName, getPriority, getStackTrace, getState, getThreadGroup, getUncaughtExceptionHandler, holdsLock, interrupt, interrupted, isAlive, isDaemon, isInterrupted, join, join, join, resume, setContextClassLoader, setDaemon, setDefaultUncaughtExceptionHandler, setName, setPriority, setUncaughtExceptionHandler, sleep, sleep, start, stop, 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
                             org.nd4j.linalg.dataset.api.MultiDataSet dataSet)

Description copied from interface: Trainer

Train on a MultiDataSet

Specified by:

feedMultiDataSet in interface Trainer

Overrides:

feedMultiDataSet in class DefaultTrainer

Parameters:

dataSet - the data set to train on

feedDataSet

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

Description copied from interface: Trainer

Train on a DataSet

Specified by:

feedDataSet in interface Trainer

Overrides:

feedDataSet in class DefaultTrainer

Parameters:

dataSet - the data set to train on

getModel

public Model getModel()

Description copied from interface: Trainer

THe current model for the trainer

Specified by:

getModel in interface Trainer

Overrides:

getModel in class DefaultTrainer

Returns:

the current Model for the worker

updateModel

public void updateModel(@NonNull
                        Model model)

Description copied from interface: Trainer

Update the current Model for the worker

Specified by:

updateModel in interface Trainer

Overrides:

updateModel in class DefaultTrainer

Parameters:

model - the new model for this worker