Package org.deeplearning4j.parallelism.parameterserver

Class ParameterServerTrainer.ParameterServerTrainerBuilder

java.lang.Object

org.deeplearning4j.parallelism.trainer.DefaultTrainer.DefaultTrainerBuilder

org.deeplearning4j.parallelism.parameterserver.ParameterServerTrainer.ParameterServerTrainerBuilder

Enclosing class:

ParameterServerTrainer

public static class ParameterServerTrainer.ParameterServerTrainerBuilder
extends DefaultTrainer.DefaultTrainerBuilder

Constructor Summary

Constructors

Constructor	Description
ParameterServerTrainerBuilder()

Method Summary

Modifier and Type	Method	Description
ParameterServerTrainer.ParameterServerTrainerBuilder	averagingFrequency(int frequency)
ParameterServerTrainer.ParameterServerTrainerBuilder	onRootModel(boolean onRootModel)
ParameterServerTrainer.ParameterServerTrainerBuilder	originalModel(org.deeplearning4j.nn.api.Model originalModel)
ParameterServerTrainer.ParameterServerTrainerBuilder	parallelWrapper(ParallelWrapper parallelWrapper)
ParameterServerTrainer.ParameterServerTrainerBuilder	queue(LinkedBlockingQueue<org.nd4j.linalg.dataset.api.DataSet> queue)
ParameterServerTrainer.ParameterServerTrainerBuilder	queueMDS(LinkedBlockingQueue<org.nd4j.linalg.dataset.api.MultiDataSet> queueMDS)
ParameterServerTrainer.ParameterServerTrainerBuilder	replicatedModel(org.deeplearning4j.nn.api.Model replicatedModel)
ParameterServerTrainer.ParameterServerTrainerBuilder	running(AtomicInteger running)
ParameterServerTrainer.ParameterServerTrainerBuilder	shouldStop(AtomicBoolean shouldStop)
ParameterServerTrainer.ParameterServerTrainerBuilder	shouldUpdate(AtomicBoolean shouldUpdate)
ParameterServerTrainer.ParameterServerTrainerBuilder	threadId(int threadId)
ParameterServerTrainer.ParameterServerTrainerBuilder	thrownException(Exception thrownException)
ParameterServerTrainer.ParameterServerTrainerBuilder	useMDS(boolean useMDS)

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

build, isStopped, lastEtlTime, nullDataSet, nullMode, toString, uuid, workspaceMode

Methods inherited from class java.lang.Object

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

Constructor Detail

ParameterServerTrainerBuilder

public ParameterServerTrainerBuilder()

Method Detail

originalModel

public ParameterServerTrainer.ParameterServerTrainerBuilder originalModel(org.deeplearning4j.nn.api.Model originalModel)

Overrides:

originalModel in class DefaultTrainer.DefaultTrainerBuilder

replicatedModel

public ParameterServerTrainer.ParameterServerTrainerBuilder replicatedModel(org.deeplearning4j.nn.api.Model replicatedModel)

Overrides:

replicatedModel in class DefaultTrainer.DefaultTrainerBuilder

queue

public ParameterServerTrainer.ParameterServerTrainerBuilder queue(LinkedBlockingQueue<org.nd4j.linalg.dataset.api.DataSet> queue)

Overrides:

queue in class DefaultTrainer.DefaultTrainerBuilder

queueMDS

public ParameterServerTrainer.ParameterServerTrainerBuilder queueMDS(LinkedBlockingQueue<org.nd4j.linalg.dataset.api.MultiDataSet> queueMDS)

Overrides:

queueMDS in class DefaultTrainer.DefaultTrainerBuilder

running

public ParameterServerTrainer.ParameterServerTrainerBuilder running(AtomicInteger running)

Overrides:

running in class DefaultTrainer.DefaultTrainerBuilder

threadId

public ParameterServerTrainer.ParameterServerTrainerBuilder threadId(int threadId)

Overrides:

threadId in class DefaultTrainer.DefaultTrainerBuilder

shouldUpdate

public ParameterServerTrainer.ParameterServerTrainerBuilder shouldUpdate(AtomicBoolean shouldUpdate)

Overrides:

shouldUpdate in class DefaultTrainer.DefaultTrainerBuilder

shouldStop

public ParameterServerTrainer.ParameterServerTrainerBuilder shouldStop(AtomicBoolean shouldStop)

Overrides:

shouldStop in class DefaultTrainer.DefaultTrainerBuilder

thrownException

public ParameterServerTrainer.ParameterServerTrainerBuilder thrownException(Exception thrownException)

Overrides:

thrownException in class DefaultTrainer.DefaultTrainerBuilder

useMDS

public ParameterServerTrainer.ParameterServerTrainerBuilder useMDS(boolean useMDS)

Overrides:

useMDS in class DefaultTrainer.DefaultTrainerBuilder

onRootModel

public ParameterServerTrainer.ParameterServerTrainerBuilder onRootModel(boolean onRootModel)

Overrides:

onRootModel in class DefaultTrainer.DefaultTrainerBuilder

parallelWrapper

public ParameterServerTrainer.ParameterServerTrainerBuilder parallelWrapper(ParallelWrapper parallelWrapper)

Overrides:

parallelWrapper in class DefaultTrainer.DefaultTrainerBuilder

averagingFrequency

public ParameterServerTrainer.ParameterServerTrainerBuilder averagingFrequency(int frequency)

Overrides:

averagingFrequency in class DefaultTrainer.DefaultTrainerBuilder