Package org.deeplearning4j.parallelism.trainer

Interface Trainer

All Superinterfaces:

Runnable

All Known Subinterfaces:

CommunicativeTrainer

All Known Implementing Classes:

DefaultTrainer, SymmetricTrainer

public interface Trainer
extends Runnable

Method Summary

Modifier and Type	Method	Description
boolean	averagingRequired()	This method returns TRUE if this Trainer implementation assumes periodic aver
void	feedDataSet(@NonNull org.nd4j.linalg.dataset.api.DataSet dataSet, long etlTime)	Train on a DataSet
void	feedMultiDataSet(@NonNull org.nd4j.linalg.dataset.api.MultiDataSet dataSet, long etlTime)	Train on a MultiDataSet
org.deeplearning4j.nn.api.Model	getModel()	THe current model for the trainer
String	getUuid()
boolean	isRunning()
void	setUncaughtExceptionHandler(Thread.UncaughtExceptionHandler handler)	Set the Thread.UncaughtExceptionHandler for this Trainer
void	shutdown()	Shutdown this worker
void	start()	Start this trainer
void	updateModel(@NonNull org.deeplearning4j.nn.api.Model model)	Update the current Model for the worker
void	updateModelParams(org.nd4j.linalg.api.ndarray.INDArray params)	This method updates replicated model params
void	updateUpdaterParams(org.nd4j.linalg.api.ndarray.INDArray params)	This method updates updater params of the replicated model
void	waitTillRunning()	Block the main thread till the trainer is up and running.

Methods inherited from interface java.lang.Runnable

run

Method Detail

feedMultiDataSet

void feedMultiDataSet(@NonNull
                      @NonNull org.nd4j.linalg.dataset.api.MultiDataSet dataSet,
                      long etlTime)

Train on a MultiDataSet

Parameters:

dataSet - the data set to train on

feedDataSet

void feedDataSet(@NonNull
                 @NonNull org.nd4j.linalg.dataset.api.DataSet dataSet,
                 long etlTime)

Train on a DataSet

Parameters:

dataSet - the data set to train on

updateModelParams

void updateModelParams(org.nd4j.linalg.api.ndarray.INDArray params)

This method updates replicated model params

Parameters:

params -

updateUpdaterParams

void updateUpdaterParams(org.nd4j.linalg.api.ndarray.INDArray params)

This method updates updater params of the replicated model

Parameters:

params -

getModel

org.deeplearning4j.nn.api.Model getModel()

THe current model for the trainer

Returns:

the current Model for the worker

updateModel

void updateModel(@NonNull
                 @NonNull org.deeplearning4j.nn.api.Model model)

Update the current Model for the worker

Parameters:

model - the new model for this worker

isRunning

boolean isRunning()

getUuid

String getUuid()

shutdown

void shutdown()

Shutdown this worker

waitTillRunning

void waitTillRunning()

Block the main thread till the trainer is up and running.

setUncaughtExceptionHandler

void setUncaughtExceptionHandler(Thread.UncaughtExceptionHandler handler)

Set the Thread.UncaughtExceptionHandler for this Trainer

Parameters:

handler - the handler for uncaught errors

start

void start()

Start this trainer

averagingRequired

boolean averagingRequired()

This method returns TRUE if this Trainer implementation assumes periodic aver

Returns: