SequentialLayers

class openff.nagl.nn.SequentialLayers(*args: Module)[source]
class openff.nagl.nn.SequentialLayers(arg: OrderedDict[str, Module])

Bases: Sequential, ContainsLayersMixin

Methods

copy

with_layers

Attributes

default_activation_function

default_dropout

training

copy(copy_weights: bool = False)[source]
default_activation_function: ClassVar[ActivationFunction] = <class 'torch.nn.modules.activation.ReLU'>
default_dropout: ClassVar[float] = 0.0
classmethod with_layers(n_input_features: int, hidden_feature_sizes: List[int], layer_activation_functions: List[ActivationFunction] | None = None, layer_dropout: List[float] | None = None)[source]