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)
def remove(module, use_out_hook=False): Norm.remove(module, RunningNorm.NAME, use_out_hook)
def remove(module, use_out_hook=True):
Norm.remove(module, BatchNorm.NAME, use_out_hook)