org.deeplearning4j.parallelism

Class ParallelWrapper

java.lang.Object

org.deeplearning4j.parallelism.ParallelWrapper

All Implemented Interfaces:

AutoCloseable

public class ParallelWrapper
extends Object
implements AutoCloseable

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	close()
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
void	fit(org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator source)
void	shutdown() This method causes all threads used for parallel training to stop

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

close

public void close()
           throws Exception

Specified by:

close in interface AutoCloseable

Throws:

Exception

shutdown

public void shutdown()

This method causes all threads used for parallel training to stop

fit

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

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 -