| Package | Description |
|---|---|
| org.deeplearning4j.parallelism |
| Modifier and Type | Method and Description |
|---|---|
ParallelWrapper.Builder |
ParallelWrapper.Builder.averageUpdaters(boolean reallyAverage)
This method enables/disables updaters averaging.
|
ParallelWrapper.Builder |
ParallelWrapper.Builder.averagingFrequency(int freq)
Model averaging frequency.
|
ParallelWrapper.Builder |
ParallelWrapper.Builder.gradientsAccumulator(org.deeplearning4j.optimize.solvers.accumulation.GradientsAccumulator accumulator)
This method allows you to specify GradientsAccumulator instance to be used in this ParallelWrapper instance
|
ParallelWrapper.Builder |
ParallelWrapper.Builder.prefetchBuffer(int size)
Size of prefetch buffer that will be used for background data prefetching.
|
ParallelWrapper.Builder |
ParallelWrapper.Builder.reportScoreAfterAveraging(boolean reallyReport)
This method enables/disables averaged model score reporting
|
ParallelWrapper.Builder |
ParallelWrapper.Builder.trainerContextArgs(Object... trainerContextArgs)
Transer context args are for calling a
TrainerContext init method
when ParallelWrapper starts training |
ParallelWrapper.Builder |
ParallelWrapper.Builder.trainerFactory(TrainerContext trainerContext)
Specify a
TrainerContext
for the given ParallelWrapper
instance. |
ParallelWrapper.Builder |
ParallelWrapper.Builder.trainingMode(ParallelWrapper.TrainingMode mode) |
ParallelWrapper.Builder |
ParallelWrapper.Builder.workers(int num)
This method allows to configure number of workers that'll be used for parallel training
|
ParallelWrapper.Builder |
ParallelWrapper.Builder.workspaceMode(org.deeplearning4j.nn.conf.WorkspaceMode mode) |
Copyright © 2017. All rights reserved.