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 and Description
ParameterServerTrainerBuilder()

Method Summary

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

Methods inherited from class java.lang.Object

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

Constructor Detail

ParameterServerTrainerBuilder

public ParameterServerTrainerBuilder()

Method Detail

originalModel

public ParameterServerTrainer.ParameterServerTrainerBuilder originalModel(Model originalModel)

replicatedModel

public ParameterServerTrainer.ParameterServerTrainerBuilder replicatedModel(Model replicatedModel)

queue

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

queueMDS

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

running

public ParameterServerTrainer.ParameterServerTrainerBuilder running(java.util.concurrent.atomic.AtomicInteger running)

threadId

public ParameterServerTrainer.ParameterServerTrainerBuilder threadId(int threadId)

shouldUpdate

public ParameterServerTrainer.ParameterServerTrainerBuilder shouldUpdate(java.util.concurrent.atomic.AtomicBoolean shouldUpdate)

shouldStop

public ParameterServerTrainer.ParameterServerTrainerBuilder shouldStop(java.util.concurrent.atomic.AtomicBoolean shouldStop)

thrownException

public ParameterServerTrainer.ParameterServerTrainerBuilder thrownException(java.lang.Exception thrownException)

useMDS

public ParameterServerTrainer.ParameterServerTrainerBuilder useMDS(boolean useMDS)

onRootModel

public ParameterServerTrainer.ParameterServerTrainerBuilder onRootModel(boolean onRootModel)

parallelWrapper

public ParameterServerTrainer.ParameterServerTrainerBuilder parallelWrapper(ParallelWrapper parallelWrapper)