torchsig.models.model_utils.simple_models.convnet_block_2d

torchsig.models.model_utils.simple_models.convnet_block_2d(in_width, out_width, kernel_shape=[5, 5], activation_fn=None)

returns a block of layers consisting of a 2d convolution, batch normalization, and activation function, with the input and output channels given by in_width and out_with, a kernel given by kernel_shape, and using the given activation_fn if no activation function is provided, this defaults to ELU