BatchNorm {fastai}

BatchNorm

Description

BatchNorm layer with 'nf' features and 'ndim' initialized depending on 'norm_type'.

Usage

BatchNorm(
  nf,
  ndim = 2,
  norm_type = 1,
  eps = 1e-05,
  momentum = 0.1,
  affine = TRUE,
  track_running_stats = TRUE
)

Arguments

nf

input shape

ndim

dimension number

norm_type

normalization type

eps

epsilon

momentum

momentum

affine

affine

track_running_stats

track running statistics

Value

None