direct.nn.conv package#
Submodules#
direct.nn.conv.conv module#
- class direct.nn.conv.conv.Conv2d(in_channels, out_channels, hidden_channels, n_convs=3, activation=PReLU(num_parameters=1), batchnorm=False)[source][source]#
Bases:
Module
Implementation of a simple cascade of 2D convolutions.
If batchnorm is set to True, batch normalization layer is applied after each convolution.