org.deeplearning4j.parallelism

Class ParallelWrapper.Builder

java.lang.Object

org.deeplearning4j.parallelism.ParallelWrapper.Builder

Enclosing class:

ParallelWrapper

public static class ParallelWrapper.Builder
extends Object

Constructor Summary

Constructors

Constructor and Description
Builder(ComputationGraph graph) Build ParallelWrapper for ComputationGraph
Builder(MultiLayerNetwork mln) Build ParallelWrapper for MultiLayerNetwork

Method Summary

Modifier and Type	Method and Description
ParallelWrapper.Builder	averageUpdaters(boolean reallyAverage) This method enables/disables updaters averaging.
ParallelWrapper.Builder	averagingFrequency(int freq) Model averaging frequency.
ParallelWrapper	build() This method returns ParallelWrapper instance
ParallelWrapper.Builder	prefetchBuffer(int size) Size of prefetch buffer that will be used for background data prefetching.
ParallelWrapper.Builder	reportScoreAfterAveraging(boolean reallyReport) This method enables/disables averaged model score reporting
ParallelWrapper.Builder	useLegacyAveraging(boolean reallyUse) If set to true, legacy averaging method is used.
ParallelWrapper.Builder	workers(int num) This method allows to configure number of workers that'll be used for parallel training

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder(@NonNull
               MultiLayerNetwork mln)

Build ParallelWrapper for MultiLayerNetwork

Parameters:

mln -

Builder

public Builder(@NonNull
               ComputationGraph graph)

Build ParallelWrapper for ComputationGraph

Parameters:

graph -

Method Detail

workers

public ParallelWrapper.Builder workers(int num)

This method allows to configure number of workers that'll be used for parallel training

Parameters:

num -

Returns:

averagingFrequency

public ParallelWrapper.Builder averagingFrequency(int freq)

Model averaging frequency.

Parameters:

freq - number of iterations between averagin

Returns:

averageUpdaters

public ParallelWrapper.Builder averageUpdaters(boolean reallyAverage)

This method enables/disables updaters averaging. Default value: TRUE PLEASE NOTE: This method is suitable for debugging purposes mostly. So don't change default value, unless you're sure why you need it.

Parameters:

reallyAverage -

Returns:

prefetchBuffer

public ParallelWrapper.Builder prefetchBuffer(int size)

Size of prefetch buffer that will be used for background data prefetching. Usually it's better to keep this value equal to the number of workers. Default value: 2

Parameters:

size - 0 to disable prefetching, any positive number

Returns:

useLegacyAveraging

public ParallelWrapper.Builder useLegacyAveraging(boolean reallyUse)

If set to true, legacy averaging method is used. This might be used as fallback on multi-gpu systems without P2P access available. Default value: false

Parameters:

reallyUse -

Returns:

reportScoreAfterAveraging

public ParallelWrapper.Builder reportScoreAfterAveraging(boolean reallyReport)

This method enables/disables averaged model score reporting

Parameters:

reallyReport -

Returns:

build

public ParallelWrapper build()

This method returns ParallelWrapper instance

Returns: