nntoolbox.vision.components.ff module

class nntoolbox.vision.components.ff.FeedforwardBlock(in_channels: int, out_features: int, pool_output_size: int, hidden_layer_sizes: Sequence = (512, ), activation: torch.nn.modules.module.Module = <class 'torch.nn.modules.activation.ReLU'>, normalization=<class 'torch.nn.modules.batchnorm.BatchNorm1d'>, bn_final: bool = False, drop_p=0.5)[source]

Bases: torch.nn.modules.container.Sequential

training: bool