org.deeplearning4j.nn.conf

Class MultiLayerConfiguration

java.lang.Object

org.deeplearning4j.nn.conf.MultiLayerConfiguration

All Implemented Interfaces:

Serializable, Cloneable

public class MultiLayerConfiguration
extends Object
implements Serializable, Cloneable

Configuration for a multi layer network

Author:

Adam Gibson

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	MultiLayerConfiguration.Builder

Field Summary

Fields

Modifier and Type	Field and Description
protected boolean	backprop
protected BackpropType	backpropType
protected CacheMode	cacheMode
protected List<NeuralNetConfiguration>	confs
protected WorkspaceMode	inferenceWorkspaceMode
protected Map<Integer,InputPreProcessor>	inputPreProcessors
protected int	iterationCount
protected boolean	pretrain
protected int	tbpttBackLength
protected int	tbpttFwdLength
protected WorkspaceMode	trainingWorkspaceMode

Constructor Summary

Constructors

Constructor and Description
MultiLayerConfiguration()

Method Summary

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)
InputPreProcessor	getInputPreProcess(int curr)
NetworkMemoryReport	getMemoryReport(InputType inputType) Get a MemoryReport for the given MultiLayerConfiguration.
String	toJson()
String	toString()
String	toYaml()

Methods inherited from class java.lang.Object

equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Field Detail

confs

protected List<NeuralNetConfiguration> confs

inputPreProcessors

protected Map<Integer,InputPreProcessor> inputPreProcessors

pretrain

protected boolean pretrain

backprop

protected boolean backprop

backpropType

protected BackpropType backpropType

tbpttFwdLength

protected int tbpttFwdLength

tbpttBackLength

protected int tbpttBackLength

trainingWorkspaceMode

protected WorkspaceMode trainingWorkspaceMode

inferenceWorkspaceMode

protected WorkspaceMode inferenceWorkspaceMode

cacheMode

protected CacheMode cacheMode

iterationCount

protected int iterationCount

Constructor Detail

MultiLayerConfiguration

public MultiLayerConfiguration()

Method Detail

toYaml

public String toYaml()

Returns:

JSON representation of NN configuration

fromYaml

public static MultiLayerConfiguration fromYaml(String json)

Create a neural net configuration from json

Parameters:

json - the neural net configuration from json

Returns:

MultiLayerConfiguration

toJson

public String toJson()

Returns:

JSON representation of NN configuration

fromJson

public static MultiLayerConfiguration fromJson(String json)

Create a neural net configuration from json

Parameters:

json - the neural net configuration from json

Returns:

MultiLayerConfiguration

toString

public String toString()

Overrides:

toString in class Object

getConf

public NeuralNetConfiguration getConf(int i)

clone

public MultiLayerConfiguration clone()

Overrides:

clone in class Object

getInputPreProcess

public InputPreProcessor getInputPreProcess(int curr)

getMemoryReport

public NetworkMemoryReport getMemoryReport(InputType inputType)

Get a MemoryReport for the given MultiLayerConfiguration. This is used to estimate the memory requirements for the given network configuration and input

Parameters:

inputType - Input types for the network

Returns:

Memory report for the network