def add(module,
affine=True,
momentum=0.9,
epsilon=1e-5,
fuse=None,
use_out_hook=True):
Norm.add(module,
RunningNorm(affine, momentum, epsilon, fuse, use_out_hook),
use_out_hook=use_out_hook)
def add(module,
affine=True,
momentum=0.9,
epsilon=1e-5,
track_statistics=True,
fuse=None,
use_out_hook=True):
Norm.add(module,
BatchNorm(affine, momentum, epsilon, track_statistics, fuse,
use_out_hook),
use_out_hook=use_out_hook)