remove_batchnorm

class torchelie.nn.utils.remove_batchnorm(m: torch.nn.modules.container.Sequential)

Remove BatchNorm in Sequentials / CondSeqs in a smart way, restoring biases in the preceding layer.