trw.layers.convs_3d

Module Contents

Functions

convs_3d(channels, convolution_kernels=(5, 5, 5), strides=(1, 1, 1), pooling_size=(2, 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')

param channels

trw.layers.convs_3d.convs_3d(channels, convolution_kernels=(5, 5, 5), strides=(1, 1, 1), pooling_size=(2, 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

  • with_lrn

  • lrn_size

  • batchnorm_momentum

  • padding (str) – if same, the convolution will be padded with zeros to keep the output shape the same as the input shape

Returns: