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	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	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:

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:

build

public ParallelWrapper build()

This method returns ParallelWrapper instance

Returns: