org.deeplearning4j.nn.transferlearning

Class FineTuneConfiguration.Builder

java.lang.Object

org.deeplearning4j.nn.transferlearning.FineTuneConfiguration.Builder

Enclosing class:

FineTuneConfiguration

public static class FineTuneConfiguration.Builder
extends Object

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
FineTuneConfiguration.Builder	activation(org.nd4j.linalg.activations.Activation activation)
FineTuneConfiguration.Builder	activation(org.nd4j.linalg.activations.IActivation activationFn)
FineTuneConfiguration.Builder	backprop(boolean backprop)
FineTuneConfiguration.Builder	backpropType(BackpropType backpropType)
FineTuneConfiguration.Builder	biasInit(double biasInit)
FineTuneConfiguration.Builder	biasUpdater(org.nd4j.linalg.learning.config.IUpdater biasUpdater)
FineTuneConfiguration	build()
FineTuneConfiguration.Builder	constraints(List<LayerConstraint> constraints)
FineTuneConfiguration.Builder	convolutionMode(ConvolutionMode convolutionMode)
FineTuneConfiguration.Builder	cudnnAlgoMode(ConvolutionLayer.AlgoMode cudnnAlgoMode)
FineTuneConfiguration.Builder	dist(Distribution dist)
FineTuneConfiguration.Builder	dropOut(double dropout)
FineTuneConfiguration.Builder	dropout(IDropout dropout)
FineTuneConfiguration.Builder	gradientNormalization(GradientNormalization gradientNormalization)
FineTuneConfiguration.Builder	gradientNormalizationThreshold(double gradientNormalizationThreshold)
FineTuneConfiguration.Builder	inferenceWorkspaceMode(WorkspaceMode inferenceWorkspaceMode)
FineTuneConfiguration.Builder	l1(double l1)
FineTuneConfiguration.Builder	l1Bias(double l1Bias)
FineTuneConfiguration.Builder	l2(double l2)
FineTuneConfiguration.Builder	l2Bias(double l2Bias)
FineTuneConfiguration.Builder	maxNumLineSearchIterations(int maxNumLineSearchIterations)
FineTuneConfiguration.Builder	miniBatch(boolean miniBatch)
FineTuneConfiguration.Builder	minimize(boolean minimize)
FineTuneConfiguration.Builder	optimizationAlgo(OptimizationAlgorithm optimizationAlgo)
FineTuneConfiguration.Builder	pretrain(boolean pretrain)
FineTuneConfiguration.Builder	seed(int seed)
FineTuneConfiguration.Builder	seed(long seed)
FineTuneConfiguration.Builder	stepFunction(StepFunction stepFunction)
FineTuneConfiguration.Builder	tbpttBackLength(int tbpttBackLength)
FineTuneConfiguration.Builder	tbpttFwdLength(int tbpttFwdLength)
String	toString()
FineTuneConfiguration.Builder	trainingWorkspaceMode(WorkspaceMode trainingWorkspaceMode)
FineTuneConfiguration.Builder	updater(org.nd4j.linalg.learning.config.IUpdater updater)
FineTuneConfiguration.Builder	updater(Updater updater) Deprecated.
FineTuneConfiguration.Builder	weightInit(WeightInit weightInit)
FineTuneConfiguration.Builder	weightNoise(IWeightNoise weightNoise)

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Method Detail

activation

public FineTuneConfiguration.Builder activation(org.nd4j.linalg.activations.IActivation activationFn)

activation

public FineTuneConfiguration.Builder activation(org.nd4j.linalg.activations.Activation activation)

weightInit

public FineTuneConfiguration.Builder weightInit(WeightInit weightInit)

biasInit

public FineTuneConfiguration.Builder biasInit(double biasInit)

dist

public FineTuneConfiguration.Builder dist(Distribution dist)

l1

public FineTuneConfiguration.Builder l1(double l1)

l2

public FineTuneConfiguration.Builder l2(double l2)

l1Bias

public FineTuneConfiguration.Builder l1Bias(double l1Bias)

l2Bias

public FineTuneConfiguration.Builder l2Bias(double l2Bias)

dropout

public FineTuneConfiguration.Builder dropout(IDropout dropout)

dropOut

public FineTuneConfiguration.Builder dropOut(double dropout)

weightNoise

public FineTuneConfiguration.Builder weightNoise(IWeightNoise weightNoise)

updater

public FineTuneConfiguration.Builder updater(org.nd4j.linalg.learning.config.IUpdater updater)

updater

@Deprecated
public FineTuneConfiguration.Builder updater(Updater updater)

Deprecated.

biasUpdater

public FineTuneConfiguration.Builder biasUpdater(org.nd4j.linalg.learning.config.IUpdater biasUpdater)

miniBatch

public FineTuneConfiguration.Builder miniBatch(boolean miniBatch)

maxNumLineSearchIterations

public FineTuneConfiguration.Builder maxNumLineSearchIterations(int maxNumLineSearchIterations)

seed

public FineTuneConfiguration.Builder seed(long seed)

seed

public FineTuneConfiguration.Builder seed(int seed)

optimizationAlgo

public FineTuneConfiguration.Builder optimizationAlgo(OptimizationAlgorithm optimizationAlgo)

stepFunction

public FineTuneConfiguration.Builder stepFunction(StepFunction stepFunction)

minimize

public FineTuneConfiguration.Builder minimize(boolean minimize)

gradientNormalization

public FineTuneConfiguration.Builder gradientNormalization(GradientNormalization gradientNormalization)

gradientNormalizationThreshold

public FineTuneConfiguration.Builder gradientNormalizationThreshold(double gradientNormalizationThreshold)

convolutionMode

public FineTuneConfiguration.Builder convolutionMode(ConvolutionMode convolutionMode)

cudnnAlgoMode

public FineTuneConfiguration.Builder cudnnAlgoMode(ConvolutionLayer.AlgoMode cudnnAlgoMode)

constraints

public FineTuneConfiguration.Builder constraints(List<LayerConstraint> constraints)

pretrain

public FineTuneConfiguration.Builder pretrain(boolean pretrain)

backprop

public FineTuneConfiguration.Builder backprop(boolean backprop)

backpropType

public FineTuneConfiguration.Builder backpropType(BackpropType backpropType)

tbpttFwdLength

public FineTuneConfiguration.Builder tbpttFwdLength(int tbpttFwdLength)

tbpttBackLength

public FineTuneConfiguration.Builder tbpttBackLength(int tbpttBackLength)

trainingWorkspaceMode

public FineTuneConfiguration.Builder trainingWorkspaceMode(WorkspaceMode trainingWorkspaceMode)

inferenceWorkspaceMode

public FineTuneConfiguration.Builder inferenceWorkspaceMode(WorkspaceMode inferenceWorkspaceMode)

build

public FineTuneConfiguration build()

toString

public String toString()

Overrides:

toString in class Object