nntoolbox.components.self_normalizing module

class nntoolbox.components.self_normalizing.SelfNormalizingMLP(in_features: int, out_features: int, hidden_layer_sizes: Sequence[int] = (512), drop_ps=(0.5, 0.5))

Bases: torch.nn.modules.container.Sequential

Implement Self-Normalizing Neural Networks: https://papers.nips.cc/paper/6698-self-normalizing-neural-networks.pdf

training: bool