org.deeplearning4j.parallelism.trainer

Interface Trainer

All Superinterfaces:

java.lang.Runnable

All Known Implementing Classes:

DefaultTrainer, ParameterServerTrainer

public interface Trainer
extends java.lang.Runnable

A Trainer is an individual worker used in ParallelWrapper for handling training in multi core situations.

Method Summary

Modifier and Type	Method and Description
void	feedDataSet(org.nd4j.linalg.dataset.api.DataSet dataSet) Train on a DataSet
void	feedMultiDataSet(org.nd4j.linalg.dataset.api.MultiDataSet dataSet) Train on a MultiDataSet
Model	getModel() THe current model for the trainer
boolean	isRunning()
void	setUncaughtExceptionHandler(java.lang.Thread.UncaughtExceptionHandler handler) Set the Thread.UncaughtExceptionHandler for this Trainer
void	shutdown() Shutdown this worker
void	start() Start this trainer
void	updateModel(Model model) Update the current Model for the worker
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
                      org.nd4j.linalg.dataset.api.MultiDataSet dataSet)

Train on a MultiDataSet

Parameters:

dataSet - the data set to train on

feedDataSet

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

Train on a DataSet

Parameters:

dataSet - the data set to train on

getModel

Model getModel()

THe current model for the trainer

Returns:

the current Model for the worker

updateModel

void updateModel(@NonNull
                 Model model)

Update the current Model for the worker

Parameters:

model - the new model for this worker

isRunning

boolean isRunning()

shutdown

void shutdown()

Shutdown this worker

waitTillRunning

void waitTillRunning()

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

setUncaughtExceptionHandler

void setUncaughtExceptionHandler(java.lang.Thread.UncaughtExceptionHandler handler)

Set the Thread.UncaughtExceptionHandler for this Trainer

Parameters:

handler - the handler for uncaught errors

start

void start()

Start this trainer