trw.layers.convs_2d
¶
Module Contents¶
Functions¶
|
|
- trw.layers.convs_2d.convs_2d(channels, convolution_kernels=(5, 5), strides=(1, 1), pooling_size=(2, 2), convolution_repeats=None, activation=nn.ReLU, with_flatten=False, dropout_probability=None, with_batchnorm=False, with_lrn=False, lrn_size=2, batchnorm_momentum=0.1, padding='same')¶
- Parameters
channels –
convolution_kernels –
strides –
pooling_size –
convolution_repeats –
activation –
with_flatten –
dropout_probability –
with_batchnorm –
batchnorm_momentum –
with_lrn –
lrn_size –
padding (str) – if same, the convolution will be padded with zeros to keep the output shape the same as the input shape
Returns: