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(@NonNull BackpropType type)
The type of backprop.
|
MultiLayerConfiguration.Builder |
MultiLayerConfiguration.Builder.cacheMode(@NonNull 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(@NonNull DataType dataType)
Set the DataType for the network parameters and activations for all layers in the network.
|
MultiLayerConfiguration.Builder |
MultiLayerConfiguration.Builder.inferenceWorkspaceMode(@NonNull 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(@NonNull 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.