Class FineTuneConfiguration
- java.lang.Object
-
- org.deeplearning4j.nn.transferlearning.FineTuneConfiguration
-
public class FineTuneConfiguration extends Object
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static class
FineTuneConfiguration.Builder
-
Field Summary
-
Constructor Summary
Constructors Constructor Description FineTuneConfiguration()
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method 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()
-
-
-
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
-
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
-
-
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)
-
-