org.deeplearning4j.nn.transferlearning

Class FineTuneConfiguration

java.lang.Object

org.deeplearning4j.nn.transferlearning.FineTuneConfiguration

public class FineTuneConfiguration
extends Object

Configuration for fine tuning. Note that values set here will override values for all non-frozen layers

Author:

Alex Black, Susan Eraly

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	FineTuneConfiguration.Builder

Field Summary

Fields

Modifier and Type	Field and Description
protected org.nd4j.linalg.activations.IActivation	activationFn
protected Boolean	backprop
protected BackpropType	backpropType
protected Double	biasInit
protected org.nd4j.linalg.learning.config.IUpdater	biasUpdater
protected org.nd4j.linalg.primitives.Optional<List<LayerConstraint>>	constraints
protected ConvolutionMode	convolutionMode
protected ConvolutionLayer.AlgoMode	cudnnAlgoMode
protected Distribution	dist
protected org.nd4j.linalg.primitives.Optional<IDropout>	dropout
protected org.nd4j.linalg.primitives.Optional<GradientNormalization>	gradientNormalization
protected Double	gradientNormalizationThreshold
protected WorkspaceMode	inferenceWorkspaceMode
protected Double	l1
protected Double	l1Bias
protected Double	l2
protected Double	l2Bias
protected Integer	maxNumLineSearchIterations
protected Boolean	miniBatch
protected Boolean	minimize
protected OptimizationAlgorithm	optimizationAlgo
protected Boolean	pretrain
protected Long	seed
protected StepFunction	stepFunction
protected Integer	tbpttBackLength
protected Integer	tbpttFwdLength
protected WorkspaceMode	trainingWorkspaceMode
protected org.nd4j.linalg.learning.config.IUpdater	updater
protected WeightInit	weightInit
protected org.nd4j.linalg.primitives.Optional<IWeightNoise>	weightNoise

Constructor Summary

Constructors

Constructor and Description
FineTuneConfiguration()

Method Summary

Modifier and Type	Method and Description
NeuralNetConfiguration	appliedNeuralNetConfiguration(NeuralNetConfiguration nnc)
NeuralNetConfiguration.Builder	appliedNeuralNetConfigurationBuilder()
void	applyToComputationGraphConfiguration(ComputationGraphConfiguration conf)
void	applyToMultiLayerConfiguration(MultiLayerConfiguration conf)
void	applyToNeuralNetConfiguration(NeuralNetConfiguration nnc)
static FineTuneConfiguration.Builder	builder()
static FineTuneConfiguration	fromJson(String json)
static FineTuneConfiguration	fromYaml(String yaml)
String	toJson()
String	toYaml()

Methods inherited from class java.lang.Object

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

Field Detail

activationFn

protected org.nd4j.linalg.activations.IActivation activationFn

weightInit

protected WeightInit weightInit

biasInit

protected Double biasInit

dist

protected Distribution dist

l1

protected Double l1

l2

protected Double l2

l1Bias

protected Double l1Bias

l2Bias

protected Double l2Bias

dropout

protected org.nd4j.linalg.primitives.Optional<IDropout> dropout

weightNoise

protected org.nd4j.linalg.primitives.Optional<IWeightNoise> weightNoise

updater

protected org.nd4j.linalg.learning.config.IUpdater updater

biasUpdater

protected org.nd4j.linalg.learning.config.IUpdater biasUpdater

miniBatch

protected Boolean miniBatch

maxNumLineSearchIterations

protected Integer maxNumLineSearchIterations

seed

protected Long seed

optimizationAlgo

protected OptimizationAlgorithm optimizationAlgo

stepFunction

protected StepFunction stepFunction

minimize

protected Boolean minimize

gradientNormalization

protected org.nd4j.linalg.primitives.Optional<GradientNormalization> gradientNormalization

gradientNormalizationThreshold

protected Double gradientNormalizationThreshold

convolutionMode

protected ConvolutionMode convolutionMode

cudnnAlgoMode

protected ConvolutionLayer.AlgoMode cudnnAlgoMode

constraints

protected org.nd4j.linalg.primitives.Optional<List<LayerConstraint>> constraints

pretrain

protected Boolean pretrain

backprop

protected Boolean backprop

backpropType

protected BackpropType backpropType

tbpttFwdLength

protected Integer tbpttFwdLength

tbpttBackLength

protected Integer tbpttBackLength

trainingWorkspaceMode

protected WorkspaceMode trainingWorkspaceMode

inferenceWorkspaceMode

protected WorkspaceMode inferenceWorkspaceMode

Constructor Detail

FineTuneConfiguration

public FineTuneConfiguration()

Method Detail

builder

public static FineTuneConfiguration.Builder builder()

appliedNeuralNetConfiguration

public NeuralNetConfiguration appliedNeuralNetConfiguration(NeuralNetConfiguration nnc)

applyToNeuralNetConfiguration

public void applyToNeuralNetConfiguration(NeuralNetConfiguration nnc)

applyToMultiLayerConfiguration

public void applyToMultiLayerConfiguration(MultiLayerConfiguration conf)

applyToComputationGraphConfiguration

public void applyToComputationGraphConfiguration(ComputationGraphConfiguration conf)

appliedNeuralNetConfigurationBuilder

public NeuralNetConfiguration.Builder appliedNeuralNetConfigurationBuilder()

toJson

public String toJson()

toYaml

public String toYaml()

fromJson

public static FineTuneConfiguration fromJson(String json)

fromYaml

public static FineTuneConfiguration fromYaml(String yaml)