| 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.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.dataType(DataType dataType)
Set the DataType for the network parameters and activations for all layers in the network.
|
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.setInputType(InputType inputType) |
MultiLayerConfiguration.Builder |
MultiLayerConfiguration.Builder.tBPTTBackwardLength(int backwardLength)
When doing truncated BPTT: how many steps of backward should we do?
|
MultiLayerConfiguration.Builder |
MultiLayerConfiguration.Builder.tBPTTForwardLength(int forwardLength)
When doing truncated BPTT: how many steps of forward pass should we do
before doing (truncated) backprop?
|
MultiLayerConfiguration.Builder |
MultiLayerConfiguration.Builder.tBPTTLength(int bpttLength)
When doing truncated BPTT: how many steps should we do?
|
MultiLayerConfiguration.Builder |
MultiLayerConfiguration.Builder.trainingWorkspaceMode(WorkspaceMode workspaceMode)
|
MultiLayerConfiguration.Builder |
MultiLayerConfiguration.Builder.validateOutputLayerConfig(boolean validate)
Enabled by default.
|
MultiLayerConfiguration.Builder |
MultiLayerConfiguration.Builder.validateTbpttConfig(boolean validate)
Enabled by default.
|
Copyright © 2020. All rights reserved.