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