org.deeplearning4j.datasets.iterator.parallel

Class JointParallelDataSetIterator.Builder

java.lang.Object

org.deeplearning4j.datasets.iterator.parallel.JointParallelDataSetIterator.Builder

Enclosing class:

JointParallelDataSetIterator

public static class JointParallelDataSetIterator.Builder
extends Object

Constructor Summary

Constructors

Constructor and Description
Builder(org.nd4j.linalg.dataset.api.iterator.enums.InequalityHandling inequalityHandling)
Builder(List<org.nd4j.linalg.dataset.api.iterator.DataSetIterator> iterators, org.nd4j.linalg.dataset.api.iterator.enums.InequalityHandling inequalityHandling)

Method Summary

Modifier and Type	Method and Description
JointParallelDataSetIterator.Builder	addSourceIterator(org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)
JointParallelDataSetIterator	build()
JointParallelDataSetIterator.Builder	enforceSingleDevice(boolean reallyEnforce)
protected boolean	hasIterator(org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)
JointParallelDataSetIterator.Builder	setBufferSizePerSplit(int bufferSize)

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder(@NonNull
               org.nd4j.linalg.dataset.api.iterator.enums.InequalityHandling inequalityHandling)

Builder

public Builder(@NonNull
               List<org.nd4j.linalg.dataset.api.iterator.DataSetIterator> iterators,
               @NonNull
               org.nd4j.linalg.dataset.api.iterator.enums.InequalityHandling inequalityHandling)

Method Detail

addSourceIterator

public JointParallelDataSetIterator.Builder addSourceIterator(@NonNull
                                                              org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)

hasIterator

protected boolean hasIterator(org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)

setBufferSizePerSplit

public JointParallelDataSetIterator.Builder setBufferSizePerSplit(int bufferSize)

enforceSingleDevice

public JointParallelDataSetIterator.Builder enforceSingleDevice(boolean reallyEnforce)

build

public JointParallelDataSetIterator build()