public class MultiLayerConfiguration extends Object implements Serializable, Cloneable
Modifier and Type | Class and Description |
---|---|
static class |
MultiLayerConfiguration.Builder |
Modifier and Type | Field and Description |
---|---|
protected BackpropType |
backpropType |
protected CacheMode |
cacheMode |
protected List<NeuralNetConfiguration> |
confs |
protected DataType |
dataType |
protected int |
epochCount |
protected WorkspaceMode |
inferenceWorkspaceMode |
protected Map<Integer,InputPreProcessor> |
inputPreProcessors |
protected int |
iterationCount |
protected int |
tbpttBackLength |
protected int |
tbpttFwdLength |
protected WorkspaceMode |
trainingWorkspaceMode |
protected boolean |
validateOutputLayerConfig |
Constructor and Description |
---|
MultiLayerConfiguration() |
Modifier and Type | Method and Description |
---|---|
MultiLayerConfiguration |
clone() |
static MultiLayerConfiguration |
fromJson(String json)
Create a neural net configuration from json
|
static MultiLayerConfiguration |
fromYaml(String json)
Create a neural net configuration from json
|
NeuralNetConfiguration |
getConf(int i) |
int |
getEpochCount() |
InputPreProcessor |
getInputPreProcess(int curr) |
List<InputType> |
getLayerActivationTypes(@NonNull InputType inputType)
For the given input shape/type for the network, return a list of activation sizes for each layer in the network.
i.e., list.get(i) is the output activation sizes for layer i |
NetworkMemoryReport |
getMemoryReport(InputType inputType)
Get a
MemoryReport for the given MultiLayerConfiguration. |
void |
setEpochCount(int epochCount) |
String |
toJson() |
String |
toString() |
String |
toYaml() |
protected List<NeuralNetConfiguration> confs
protected Map<Integer,InputPreProcessor> inputPreProcessors
protected BackpropType backpropType
protected int tbpttFwdLength
protected int tbpttBackLength
protected boolean validateOutputLayerConfig
protected WorkspaceMode trainingWorkspaceMode
protected WorkspaceMode inferenceWorkspaceMode
protected CacheMode cacheMode
protected DataType dataType
protected int iterationCount
protected int epochCount
public int getEpochCount()
public void setEpochCount(int epochCount)
public String toYaml()
public static MultiLayerConfiguration fromYaml(String json)
json
- the neural net configuration from jsonMultiLayerConfiguration
public String toJson()
public static MultiLayerConfiguration fromJson(String json)
json
- the neural net configuration from jsonMultiLayerConfiguration
public NeuralNetConfiguration getConf(int i)
public MultiLayerConfiguration clone()
public InputPreProcessor getInputPreProcess(int curr)
public NetworkMemoryReport getMemoryReport(InputType inputType)
MemoryReport
for the given MultiLayerConfiguration. This is used to estimate the
memory requirements for the given network configuration and inputinputType
- Input types for the networkpublic List<InputType> getLayerActivationTypes(@NonNull @NonNull InputType inputType)
inputType
- Input type for the networkCopyright © 2021. All rights reserved.