speechbrain.lobes.models.VanillaNN module
Vanilla Neural Network for simple tests.
Authors * Elena Rastorgueva 2020
Summary
Classes:
A simple vanilla Deep Neural Network. |
Reference
- class speechbrain.lobes.models.VanillaNN.VanillaNN(input_shape, activation=<class 'torch.nn.modules.activation.LeakyReLU'>, dnn_blocks=2, dnn_neurons=512)[source]
Bases:
Sequential
A simple vanilla Deep Neural Network.
- Parameters:
Example
>>> inputs = torch.rand([10, 120, 60]) >>> model = VanillaNN(input_shape=inputs.shape) >>> outputs = model(inputs) >>> outputs.shape torch.Size([10, 120, 512])