org.deeplearning4j.parallelism

Class EarlyStoppingParallelTrainer<T extends org.deeplearning4j.nn.api.Model>

java.lang.Object

org.deeplearning4j.parallelism.EarlyStoppingParallelTrainer<T>

All Implemented Interfaces:

org.deeplearning4j.earlystopping.trainer.IEarlyStoppingTrainer<T>

public class EarlyStoppingParallelTrainer<T extends org.deeplearning4j.nn.api.Model>
extends Object
implements org.deeplearning4j.earlystopping.trainer.IEarlyStoppingTrainer<T>

Conduct parallel early stopping training with ParallelWrapper under the hood.
Can be used to train a MultiLayerNetwork or a ComputationGraph via early stopping.

Author:

Justin Long (crockpotveggies)

Field Summary

Fields

Modifier and Type	Field and Description
protected org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<T>	esConfig
protected T	model
protected org.deeplearning4j.earlystopping.termination.IterationTerminationCondition	terminationReason

Constructor Summary

Constructors

Constructor and Description
EarlyStoppingParallelTrainer(org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<T> earlyStoppingConfiguration, T model, org.nd4j.linalg.dataset.api.iterator.DataSetIterator train, org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator trainMulti, org.deeplearning4j.earlystopping.listener.EarlyStoppingListener<T> listener, int workers, int prefetchBuffer, int averagingFrequency)
EarlyStoppingParallelTrainer(org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<T> earlyStoppingConfiguration, T model, org.nd4j.linalg.dataset.api.iterator.DataSetIterator train, org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator trainMulti, org.deeplearning4j.earlystopping.listener.EarlyStoppingListener<T> listener, int workers, int prefetchBuffer, int averagingFrequency, boolean reportScoreAfterAveraging, boolean useLegacyAveraging)
EarlyStoppingParallelTrainer(org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<T> earlyStoppingConfiguration, T model, org.nd4j.linalg.dataset.api.iterator.DataSetIterator train, org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator trainMulti, int workers, int prefetchBuffer, int averagingFrequency)

Method Summary

Modifier and Type	Method and Description
org.deeplearning4j.earlystopping.EarlyStoppingResult<T>	fit()
boolean	getTermination()
void	incrementIteration()
protected void	reset()
void	setLatestScore(double latestScore)
void	setListener(org.deeplearning4j.earlystopping.listener.EarlyStoppingListener<T> listener)
void	setTermination(boolean terminate)
protected void	setTerminationReason(org.deeplearning4j.earlystopping.termination.IterationTerminationCondition terminationReason)

Methods inherited from class java.lang.Object

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

Field Detail

model

protected T extends org.deeplearning4j.nn.api.Model model

esConfig

protected final org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<T extends org.deeplearning4j.nn.api.Model> esConfig

terminationReason

protected volatile org.deeplearning4j.earlystopping.termination.IterationTerminationCondition terminationReason

Constructor Detail

EarlyStoppingParallelTrainer

public EarlyStoppingParallelTrainer(org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<T> earlyStoppingConfiguration,
                                    T model,
                                    org.nd4j.linalg.dataset.api.iterator.DataSetIterator train,
                                    org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator trainMulti,
                                    int workers,
                                    int prefetchBuffer,
                                    int averagingFrequency)

EarlyStoppingParallelTrainer

public EarlyStoppingParallelTrainer(org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<T> earlyStoppingConfiguration,
                                    T model,
                                    org.nd4j.linalg.dataset.api.iterator.DataSetIterator train,
                                    org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator trainMulti,
                                    org.deeplearning4j.earlystopping.listener.EarlyStoppingListener<T> listener,
                                    int workers,
                                    int prefetchBuffer,
                                    int averagingFrequency)

EarlyStoppingParallelTrainer

public EarlyStoppingParallelTrainer(org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<T> earlyStoppingConfiguration,
                                    T model,
                                    org.nd4j.linalg.dataset.api.iterator.DataSetIterator train,
                                    org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator trainMulti,
                                    org.deeplearning4j.earlystopping.listener.EarlyStoppingListener<T> listener,
                                    int workers,
                                    int prefetchBuffer,
                                    int averagingFrequency,
                                    boolean reportScoreAfterAveraging,
                                    boolean useLegacyAveraging)

Method Detail

setTerminationReason

protected void setTerminationReason(org.deeplearning4j.earlystopping.termination.IterationTerminationCondition terminationReason)

fit

public org.deeplearning4j.earlystopping.EarlyStoppingResult<T> fit()

Specified by:

fit in interface org.deeplearning4j.earlystopping.trainer.IEarlyStoppingTrainer<T extends org.deeplearning4j.nn.api.Model>

setLatestScore

public void setLatestScore(double latestScore)

incrementIteration

public void incrementIteration()

setTermination

public void setTermination(boolean terminate)

getTermination

public boolean getTermination()

setListener

public void setListener(org.deeplearning4j.earlystopping.listener.EarlyStoppingListener<T> listener)

Specified by:

setListener in interface org.deeplearning4j.earlystopping.trainer.IEarlyStoppingTrainer<T extends org.deeplearning4j.nn.api.Model>

reset

protected void reset()