model.dynamic

Attributes

mlp_example

Classes

DynamicNN

Class that enables the instantiation of custom neural network architectures using a list

Module Contents

class model.dynamic.DynamicNN(config: List[Dict], custom_layers: List | None = None)

Bases: torch.nn.Sequential

Class that enables the instantiation of custom neural network architectures using a list of layer configurations.

Inherits from:

torch.nn.Sequential: Parent class for implementing neural networks with modules defined in a sequential manner.

Initializes a DynamicNN object.

Parameters:

  • config (list[dict]) – List of layer configurations. Each configuration is a dictionary specifying the layer type and its corresponding parameters.

  • custom_layers (list, optional) – List of custom layer classes. Defaults to None.

model.dynamic.mlp_example