Package | Description |
---|---|
org.deeplearning4j.nn.conf |
Modifier and Type | Class and Description |
---|---|
static class |
NeuralNetConfiguration.ListBuilder
Fluent interface for building a list of configurations
|
Modifier and Type | Method and Description |
---|---|
MultiLayerConfiguration.Builder |
MultiLayerConfiguration.Builder.backprop(boolean backprop)
Whether to do back prop or not
|
MultiLayerConfiguration.Builder |
MultiLayerConfiguration.Builder.backpropType(BackpropType type)
The type of backprop.
|
MultiLayerConfiguration.Builder |
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 |
MultiLayerConfiguration.Builder.confs(List<NeuralNetConfiguration> confs) |
MultiLayerConfiguration.Builder |
MultiLayerConfiguration.Builder.inferenceWorkspaceMode(WorkspaceMode workspaceMode)
|
MultiLayerConfiguration.Builder |
MultiLayerConfiguration.Builder.inputPreProcessor(Integer layer,
InputPreProcessor processor)
Specify the processors.
|
MultiLayerConfiguration.Builder |
MultiLayerConfiguration.Builder.inputPreProcessors(Map<Integer,InputPreProcessor> processors) |
MultiLayerConfiguration.Builder |
MultiLayerConfiguration.Builder.pretrain(boolean pretrain)
Whether to do pre train or not
|
MultiLayerConfiguration.Builder |
MultiLayerConfiguration.Builder.setInputType(InputType inputType) |
MultiLayerConfiguration.Builder |
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 |
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 |
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 |
MultiLayerConfiguration.Builder.trainingWorkspaceMode(WorkspaceMode workspaceMode)
|
Copyright © 2018. All rights reserved.