peal.architectures.interfaces.VGGConfig¶

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

Bases: BaseModel

The config template for a VGG Layer.

Methods

Attributes

`config_name`	The type of config.
`num_blocks`	Number of blocks per layer.
`num_neurons`	The number of neurons in the layer.
`receptive_field`	The size of the receptive field.
`tensor_dim`	The dimension of the tensor.
`use_batchnorm`	Whether to use batchnorm or not.

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

num_neurons: pydantic.PositiveInt = 512¶: The number of neurons in the layer.

num_blocks: pydantic.PositiveInt = 2¶: Number of blocks per layer.

use_batchnorm: bool = True¶: Whether to use batchnorm or not.

receptive_field: pydantic.PositiveInt = 3¶: The size of the receptive field.

tensor_dim: pydantic.PositiveInt = 2¶: The dimension of the tensor. Options: [1, 2, 3]