org.deeplearning4j.datasets.iterator

Class AsyncMultiDataSetIterator

java.lang.Object

org.deeplearning4j.datasets.iterator.AsyncMultiDataSetIterator

All Implemented Interfaces:

Serializable, Iterator<org.nd4j.linalg.dataset.api.MultiDataSet>, org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator

public class AsyncMultiDataSetIterator
extends Object
implements org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator

Async prefetching iterator wrapper for MultiDataSetIterator implementations

PLEASE NOTE: If used together with CUDA backend, please use it with caution.

Author:

Alex Black, [email protected]

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
AsyncMultiDataSetIterator(org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator iterator, int queueLength)

Method Summary

Modifier and Type	Method and Description
boolean	asyncSupported()
boolean	hasNext()
org.nd4j.linalg.dataset.api.MultiDataSet	next()
org.nd4j.linalg.dataset.api.MultiDataSet	next(int num)
void	remove()
void	reset()
boolean	resetSupported()
void	setPreProcessor(org.nd4j.linalg.dataset.api.MultiDataSetPreProcessor preProcessor)
void	shutdown() Shut down the async data set iterator thread This is not typically necessary if using a single AsyncDataSetIterator (thread is a daemon thread and so shouldn't block the JVM from exiting) Behaviour of next(), hasNext() etc methods after shutdown of async iterator is undefined

Methods inherited from class java.lang.Object

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

Methods inherited from interface java.util.Iterator

forEachRemaining

Constructor Detail

AsyncMultiDataSetIterator

public AsyncMultiDataSetIterator(org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator iterator,
                                 int queueLength)

Method Detail

next

public org.nd4j.linalg.dataset.api.MultiDataSet next(int num)

Specified by:

next in interface org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator

setPreProcessor

public void setPreProcessor(org.nd4j.linalg.dataset.api.MultiDataSetPreProcessor preProcessor)

Specified by:

setPreProcessor in interface org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator

resetSupported

public boolean resetSupported()

Specified by:

resetSupported in interface org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator

asyncSupported

public boolean asyncSupported()

Specified by:

asyncSupported in interface org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator

reset

public void reset()

Specified by:

reset in interface org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator

hasNext

public boolean hasNext()

Specified by:

hasNext in interface Iterator<org.nd4j.linalg.dataset.api.MultiDataSet>

next

public org.nd4j.linalg.dataset.api.MultiDataSet next()

Specified by:

next in interface Iterator<org.nd4j.linalg.dataset.api.MultiDataSet>

remove

public void remove()

Specified by:

remove in interface Iterator<org.nd4j.linalg.dataset.api.MultiDataSet>

shutdown

public void shutdown()

Shut down the async data set iterator thread This is not typically necessary if using a single AsyncDataSetIterator (thread is a daemon thread and so shouldn't block the JVM from exiting) Behaviour of next(), hasNext() etc methods after shutdown of async iterator is undefined