org.deeplearning4j.spark.parameterserver

Class ParameterServerTrainingHook

java.lang.Object

org.deeplearning4j.spark.parameterserver.ParameterServerTrainingHook

All Implemented Interfaces:

java.io.Serializable, TrainingHook

public class ParameterServerTrainingHook
extends java.lang.Object
implements TrainingHook

Training hook for the parameter server

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
ParameterServerTrainingHook()

Method Summary

Modifier and Type	Method and Description
void	postUpdate(org.nd4j.linalg.dataset.api.DataSet minibatch, Model model) A hook method for post update
void	postUpdate(org.nd4j.linalg.dataset.api.MultiDataSet minibatch, Model model) A hook method for post update
void	preUpdate(org.nd4j.linalg.dataset.api.DataSet minibatch, Model model) A hook method for pre update.
void	preUpdate(org.nd4j.linalg.dataset.api.MultiDataSet minibatch, Model model) A hook method for pre update.

Methods inherited from class java.lang.Object

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

Constructor Detail

ParameterServerTrainingHook

public ParameterServerTrainingHook()

Method Detail

preUpdate

public void preUpdate(org.nd4j.linalg.dataset.api.DataSet minibatch,
                      Model model)

A hook method for pre update.

Specified by:

preUpdate in interface TrainingHook

Parameters:

minibatch - the inibatch that was used for the update

model - themodel that was update

postUpdate

public void postUpdate(org.nd4j.linalg.dataset.api.DataSet minibatch,
                       Model model)

A hook method for post update

Specified by:

postUpdate in interface TrainingHook

Parameters:

minibatch - the minibatch that was usd for the update

model - the model that was updated

preUpdate

public void preUpdate(org.nd4j.linalg.dataset.api.MultiDataSet minibatch,
                      Model model)

A hook method for pre update.

Specified by:

preUpdate in interface TrainingHook

Parameters:

minibatch - the inibatch that was used for the update

model - themodel that was update

postUpdate

public void postUpdate(org.nd4j.linalg.dataset.api.MultiDataSet minibatch,
                       Model model)

A hook method for post update

Specified by:

postUpdate in interface TrainingHook

Parameters:

minibatch - the minibatch that was usd for the update

model - the model that was updated