org.deeplearning4j.nn.transferlearning

Class FineTuneConfiguration

java.lang.Object

org.deeplearning4j.nn.transferlearning.FineTuneConfiguration

public class FineTuneConfiguration
extends Object

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 IActivation	activationFn
protected Boolean	backprop
protected BackpropType	backpropType
protected Double	biasInit
protected IUpdater	biasUpdater
protected Optional<List<LayerConstraint>>	constraints
protected ConvolutionMode	convolutionMode
protected ConvolutionLayer.AlgoMode	cudnnAlgoMode
protected Optional<IDropout>	dropout
protected Optional<GradientNormalization>	gradientNormalization
protected Double	gradientNormalizationThreshold
protected WorkspaceMode	inferenceWorkspaceMode
protected Integer	maxNumLineSearchIterations
protected Boolean	miniBatch
protected Boolean	minimize
protected OptimizationAlgorithm	optimizationAlgo
protected Boolean	pretrain
protected List<Regularization>	regularization
protected List<Regularization>	regularizationBias
protected boolean	removeL1
protected boolean	removeL1Bias
protected boolean	removeL2
protected boolean	removeL2Bias
protected boolean	removeWD
protected boolean	removeWDBias
protected Long	seed
protected StepFunction	stepFunction
protected Integer	tbpttBackLength
protected Integer	tbpttFwdLength
protected WorkspaceMode	trainingWorkspaceMode
protected IUpdater	updater
protected IWeightInit	weightInitFn
protected 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 IActivation activationFn

weightInitFn

protected IWeightInit weightInitFn

biasInit

protected Double biasInit

regularization

protected List<Regularization> regularization

regularizationBias

protected List<Regularization> regularizationBias

removeL2

protected boolean removeL2

removeL2Bias

protected boolean removeL2Bias

removeL1

protected boolean removeL1

removeL1Bias

protected boolean removeL1Bias

removeWD

protected boolean removeWD

removeWDBias

protected boolean removeWDBias

dropout

protected Optional<IDropout> dropout

weightNoise

protected Optional<IWeightNoise> weightNoise

updater

protected IUpdater updater

biasUpdater

protected 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 Optional<GradientNormalization> gradientNormalization

gradientNormalizationThreshold

protected Double gradientNormalizationThreshold

convolutionMode

protected ConvolutionMode convolutionMode

cudnnAlgoMode

protected ConvolutionLayer.AlgoMode cudnnAlgoMode

constraints

protected 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)