public static class MultiLayerConfiguration.Builder extends Object
Modifier and Type | Field and Description |
---|---|
protected boolean |
backprop |
protected BackpropType |
backpropType |
protected CacheMode |
cacheMode |
protected List<NeuralNetConfiguration> |
confs |
protected double |
dampingFactor |
protected WorkspaceMode |
inferenceWorkspaceMode |
protected Map<Integer,InputPreProcessor> |
inputPreProcessors |
protected InputType |
inputType |
protected boolean |
pretrain |
protected int |
tbpttBackLength |
protected int |
tbpttFwdLength |
protected WorkspaceMode |
trainingWorkspaceMode |
Constructor and Description |
---|
Builder() |
Modifier and Type | Method and Description |
---|---|
MultiLayerConfiguration.Builder |
backprop(boolean backprop)
Whether to do back prop or not
|
MultiLayerConfiguration.Builder |
backpropType(BackpropType type)
The type of backprop.
|
MultiLayerConfiguration |
build() |
MultiLayerConfiguration.Builder |
cacheMode(CacheMode cacheMode)
This method defines how/if preOutput cache is handled:
NONE: cache disabled (default value)
HOST: Host memory will be used
DEVICE: GPU memory will be used (on CPU backends effect will be the same as for HOST)
|
MultiLayerConfiguration.Builder |
confs(List<NeuralNetConfiguration> confs) |
MultiLayerConfiguration.Builder |
inferenceWorkspaceMode(WorkspaceMode workspaceMode)
This method defines Workspace mode being used during inference:
NONE: workspace won't be used
SINGLE: one workspace will be used during whole iteration loop
SEPARATE: separate workspaces will be used for feedforward and backprop iteration loops
|
MultiLayerConfiguration.Builder |
inputPreProcessor(Integer layer,
InputPreProcessor processor)
Specify the processors.
|
MultiLayerConfiguration.Builder |
inputPreProcessors(Map<Integer,InputPreProcessor> processors) |
MultiLayerConfiguration.Builder |
pretrain(boolean pretrain)
Whether to do pre train or not
|
MultiLayerConfiguration.Builder |
setInputType(InputType inputType) |
MultiLayerConfiguration.Builder |
tBPTTBackwardLength(int backwardLength)
When doing truncated BPTT: how many steps of backward should we do?
Only applicable when doing backpropType(BackpropType.TruncatedBPTT) This is the k2 parameter on pg23 of http://www.cs.utoronto.ca/~ilya/pubs/ilya_sutskever_phd_thesis.pdf |
MultiLayerConfiguration.Builder |
tBPTTForwardLength(int forwardLength)
When doing truncated BPTT: how many steps of forward pass should we do
before doing (truncated) backprop?
Only applicable when doing backpropType(BackpropType.TruncatedBPTT) Typically tBPTTForwardLength parameter is same as the tBPTTBackwardLength parameter, but may be larger than it in some circumstances (but never smaller) Ideally your training data time series length should be divisible by this This is the k1 parameter on pg23 of http://www.cs.utoronto.ca/~ilya/pubs/ilya_sutskever_phd_thesis.pdf |
MultiLayerConfiguration.Builder |
tBPTTLength(int bpttLength)
When doing truncated BPTT: how many steps should we do?
Only applicable when doing backpropType(BackpropType.TruncatedBPTT) See: http://www.cs.utoronto.ca/~ilya/pubs/ilya_sutskever_phd_thesis.pdf |
MultiLayerConfiguration.Builder |
trainingWorkspaceMode(WorkspaceMode workspaceMode)
This method defines Workspace mode being used during training:
NONE: workspace won't be used
SINGLE: one workspace will be used during whole iteration loop
SEPARATE: separate workspaces will be used for feedforward and backprop iteration loops
|
protected List<NeuralNetConfiguration> confs
protected double dampingFactor
protected Map<Integer,InputPreProcessor> inputPreProcessors
protected boolean pretrain
protected boolean backprop
protected BackpropType backpropType
protected int tbpttFwdLength
protected int tbpttBackLength
protected InputType inputType
protected WorkspaceMode trainingWorkspaceMode
protected WorkspaceMode inferenceWorkspaceMode
protected CacheMode cacheMode
public MultiLayerConfiguration.Builder inputPreProcessor(Integer layer, InputPreProcessor processor)
processor
- what to use to preProcess the data.public MultiLayerConfiguration.Builder inputPreProcessors(Map<Integer,InputPreProcessor> processors)
public MultiLayerConfiguration.Builder backprop(boolean backprop)
backprop
- whether to do back prop or notpublic MultiLayerConfiguration.Builder trainingWorkspaceMode(@NonNull WorkspaceMode workspaceMode)
workspaceMode
- public MultiLayerConfiguration.Builder inferenceWorkspaceMode(@NonNull WorkspaceMode workspaceMode)
workspaceMode
- public MultiLayerConfiguration.Builder cacheMode(@NonNull CacheMode cacheMode)
cacheMode
- public MultiLayerConfiguration.Builder backpropType(BackpropType type)
public MultiLayerConfiguration.Builder tBPTTLength(int bpttLength)
bpttLength
- length > 0public MultiLayerConfiguration.Builder tBPTTForwardLength(int forwardLength)
forwardLength
- Forward length > 0, >= backwardLengthpublic MultiLayerConfiguration.Builder tBPTTBackwardLength(int backwardLength)
backwardLength
- <= forwardLengthpublic MultiLayerConfiguration.Builder pretrain(boolean pretrain)
pretrain
- whether to do pre train or notpublic MultiLayerConfiguration.Builder confs(List<NeuralNetConfiguration> confs)
public MultiLayerConfiguration.Builder setInputType(InputType inputType)
public MultiLayerConfiguration build()
Copyright © 2017. All rights reserved.