org.deeplearning4j.parallelism

Class ParallelWrapper

java.lang.Object

org.deeplearning4j.parallelism.ParallelWrapper

public class ParallelWrapper
extends Object

This is simple data-parallel wrapper suitable for multi-cpu/multi-gpu environments.

Author:

[email protected]

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	ParallelWrapper.Builder

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	ParallelWrapper(Model model, int workers, int prefetchSize)

Method Summary

Modifier and Type	Method and Description
void	fit(org.nd4j.linalg.dataset.api.iterator.DataSetIterator source) This method takes DataSetIterator, and starts training over it by scheduling DataSets to different executors

Methods inherited from class java.lang.Object

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

Constructor Detail

ParallelWrapper

protected ParallelWrapper(Model model,
                          int workers,
                          int prefetchSize)

Method Detail

fit

public void fit(@NonNull
                org.nd4j.linalg.dataset.api.iterator.DataSetIterator source)

This method takes DataSetIterator, and starts training over it by scheduling DataSets to different executors

Parameters:

source -