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
-
Methods inherited from class org.deeplearning4j.parallelism.trainer.DefaultTrainer.DefaultTrainerBuilder
build, isStopped, lastEtlTime, nullDataSet, nullMode, toString, uuid, workspaceMode
-
-
-
-
Method Detail
-
originalModel
public ParameterServerTrainer.ParameterServerTrainerBuilder originalModel(org.deeplearning4j.nn.api.Model originalModel)
- Overrides:
originalModelin classDefaultTrainer.DefaultTrainerBuilder
-
replicatedModel
public ParameterServerTrainer.ParameterServerTrainerBuilder replicatedModel(org.deeplearning4j.nn.api.Model replicatedModel)
- Overrides:
replicatedModelin classDefaultTrainer.DefaultTrainerBuilder
-
queue
public ParameterServerTrainer.ParameterServerTrainerBuilder queue(LinkedBlockingQueue<org.nd4j.linalg.dataset.api.DataSet> queue)
- Overrides:
queuein classDefaultTrainer.DefaultTrainerBuilder
-
queueMDS
public ParameterServerTrainer.ParameterServerTrainerBuilder queueMDS(LinkedBlockingQueue<org.nd4j.linalg.dataset.api.MultiDataSet> queueMDS)
- Overrides:
queueMDSin classDefaultTrainer.DefaultTrainerBuilder
-
running
public ParameterServerTrainer.ParameterServerTrainerBuilder running(AtomicInteger running)
- Overrides:
runningin classDefaultTrainer.DefaultTrainerBuilder
-
threadId
public ParameterServerTrainer.ParameterServerTrainerBuilder threadId(int threadId)
- Overrides:
threadIdin classDefaultTrainer.DefaultTrainerBuilder
-
shouldUpdate
public ParameterServerTrainer.ParameterServerTrainerBuilder shouldUpdate(AtomicBoolean shouldUpdate)
- Overrides:
shouldUpdatein classDefaultTrainer.DefaultTrainerBuilder
-
shouldStop
public ParameterServerTrainer.ParameterServerTrainerBuilder shouldStop(AtomicBoolean shouldStop)
- Overrides:
shouldStopin classDefaultTrainer.DefaultTrainerBuilder
-
thrownException
public ParameterServerTrainer.ParameterServerTrainerBuilder thrownException(Exception thrownException)
- Overrides:
thrownExceptionin classDefaultTrainer.DefaultTrainerBuilder
-
useMDS
public ParameterServerTrainer.ParameterServerTrainerBuilder useMDS(boolean useMDS)
- Overrides:
useMDSin classDefaultTrainer.DefaultTrainerBuilder
-
onRootModel
public ParameterServerTrainer.ParameterServerTrainerBuilder onRootModel(boolean onRootModel)
- Overrides:
onRootModelin classDefaultTrainer.DefaultTrainerBuilder
-
parallelWrapper
public ParameterServerTrainer.ParameterServerTrainerBuilder parallelWrapper(ParallelWrapper parallelWrapper)
- Overrides:
parallelWrapperin classDefaultTrainer.DefaultTrainerBuilder
-
averagingFrequency
public ParameterServerTrainer.ParameterServerTrainerBuilder averagingFrequency(int frequency)
- Overrides:
averagingFrequencyin classDefaultTrainer.DefaultTrainerBuilder
-
-