Introduction

Tutorials

API Reference

public conv_bnfunction

conv_bn(kernelsize, inplanes, outplanes, activation = relu;
        rev = false, preact = true,
        stride = 1, pad = 0, dilation = 1, groups = 1, [bias, weight, init],
        initβ = Flux.zeros32, initγ = Flux.ones32, ϵ = 1f-5, momentum = 1f-1)

Create a convolution + batch normalization pair with activation.

Arguments

  • kernelsize: size of the convolution kernel (tuple)
  • inplanes: number of input feature maps
  • outplanes: number of output feature maps
  • activation: the activation function for the final layer
  • rev: set to true to place the batch norm before the convolution
  • preact: set to true to place the activation function before the batch norm (only compatible with rev = false)
  • stride: stride of the convolution kernel
  • pad: padding of the convolution kernel
  • dilation: dilation of the convolution kernel
  • groups: groups for the convolution kernel
  • bias, weight, init: initialization for the convolution kernel (see Flux.Conv)
  • initβ, initγ: initialization for the batch norm (see Flux.BatchNorm)
  • ϵ, momentum: batch norm parameters (see Flux.BatchNorm)