org.deeplearning4j.nn.graph.vertex.impl

Class FrozenVertex

java.lang.Object

org.deeplearning4j.nn.graph.vertex.BaseWrapperVertex

org.deeplearning4j.nn.graph.vertex.impl.FrozenVertex

All Implemented Interfaces:

Serializable, Trainable, GraphVertex

public class FrozenVertex
extends BaseWrapperVertex

FrozenVertex is used for the purposes of transfer learning A frozen layers wraps another DL4J GraphVertex within it. During backprop, the FrozenVertex is skipped, and any parameters are not be updated.

Author:

Alex Black

See Also:

Serialized Form

Field Summary

Fields inherited from class org.deeplearning4j.nn.graph.vertex.BaseWrapperVertex

underlying

Constructor Summary

Constructors

Constructor and Description
FrozenVertex(GraphVertex underlying)

Method Summary

Modifier and Type	Method and Description
TrainingConfig	getConfig()

Methods inherited from class org.deeplearning4j.nn.graph.vertex.BaseWrapperVertex

canDoBackward, canDoForward, clear, clearVertex, doBackward, doForward, feedForwardMaskArrays, getEpsilon, getGradientsViewArray, getInputs, getInputVertices, getLayer, getNumInputArrays, getNumOutputConnections, getOutputVertices, getVertexIndex, getVertexName, hasLayer, isInputVertex, isOutputVertex, numParams, params, paramTable, setBackpropGradientsViewArray, setEpsilon, setInput, setInputs, setInputVertices, setLayerAsFrozen, setOutputVertex, setOutputVertices, updaterDivideByMinibatch

Methods inherited from class java.lang.Object

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

Constructor Detail

FrozenVertex

public FrozenVertex(GraphVertex underlying)

Method Detail

getConfig

public TrainingConfig getConfig()

Specified by:

getConfig in interface Trainable

Overrides:

getConfig in class BaseWrapperVertex

Returns:

Training configuration