org.deeplearning4j.datasets.iterator.callbacks

Class InterleavedDataSetCallback

java.lang.Object

org.deeplearning4j.datasets.iterator.callbacks.InterleavedDataSetCallback

All Implemented Interfaces:

DataSetCallback

public class InterleavedDataSetCallback
extends Object
implements DataSetCallback

This callback migrates incoming datasets in round-robin manner, to ensure TDA for ParallelWrapper

Author:

[email protected]

Constructor Summary

Constructors

Constructor and Description
InterleavedDataSetCallback(int bufferSize)

Method Summary

Modifier and Type	Method and Description
void	call(org.nd4j.linalg.dataset.api.DataSet dataSet)
void	call(org.nd4j.linalg.dataset.api.MultiDataSet multiDataSet)
protected void	initializeWorkspaces(long size)
void	reset()

Methods inherited from class java.lang.Object

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

Constructor Detail

InterleavedDataSetCallback

public InterleavedDataSetCallback(int bufferSize)

Method Detail

initializeWorkspaces

protected void initializeWorkspaces(long size)

call

public void call(org.nd4j.linalg.dataset.api.DataSet dataSet)

Specified by:

call in interface DataSetCallback

call

public void call(org.nd4j.linalg.dataset.api.MultiDataSet multiDataSet)

Specified by:

call in interface DataSetCallback

reset

public void reset()

Specified by:

reset in interface DataSetCallback