def remove(module, use_out_hook=True): Norm.remove(module, BatchNorm.NAME, use_out_hook)
def remove(module, use_out_hook=False): Norm.remove(module, RunningNorm.NAME, use_out_hook)