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)