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¶
-