org.deeplearning4j.spark.impl.paramavg.aggregator

Class ParameterAveragingElementCombineFunction

java.lang.Object

org.deeplearning4j.spark.impl.paramavg.aggregator.ParameterAveragingElementCombineFunction

All Implemented Interfaces:

java.io.Serializable, org.apache.spark.api.java.function.Function2<ParameterAveragingAggregationTuple,ParameterAveragingAggregationTuple,ParameterAveragingAggregationTuple>

public class ParameterAveragingElementCombineFunction
extends java.lang.Object
implements org.apache.spark.api.java.function.Function2<ParameterAveragingAggregationTuple,ParameterAveragingAggregationTuple,ParameterAveragingAggregationTuple>

Function used in ParameterAveraging TrainingMaster, for doing parameter averaging, and handling updaters

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
ParameterAveragingElementCombineFunction()

Method Summary

Modifier and Type	Method and Description
ParameterAveragingAggregationTuple	call(ParameterAveragingAggregationTuple v1, ParameterAveragingAggregationTuple v2)

Methods inherited from class java.lang.Object

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

Constructor Detail

ParameterAveragingElementCombineFunction

public ParameterAveragingElementCombineFunction()

Method Detail

call

public ParameterAveragingAggregationTuple call(ParameterAveragingAggregationTuple v1,
                                               ParameterAveragingAggregationTuple v2)
                                        throws java.lang.Exception

Specified by:

call in interface org.apache.spark.api.java.function.Function2<ParameterAveragingAggregationTuple,ParameterAveragingAggregationTuple,ParameterAveragingAggregationTuple>

Throws:

java.lang.Exception