peal.architectures.interfaces.ArchitectureConfig¶

class peal.architectures.interfaces.ArchitectureConfig[source]¶

Bases: BaseModel

The config template for a neural architecture.

Methods

Attributes

`activation`	The activation function used in the architecture.
`config_name`	The type of config.
`layers`	The layers of the architecture.

config_name: str = 'ArchitectureConfig'¶: The type of config. This is necessary to find config class from yaml config

layers: list¶: The layers of the architecture. Elements of the list are tuples of the form (layer_type, *layer_config). Options for list elements: [‘fc’, ‘vgg’,’resnet’,’transformer’]

activation: str = 'ReLU'¶: The activation function used in the architecture. Options: [‘ReLU’, ‘LeakyReLU’, ‘LeakySoftplus’]