Pytorch Utils
Algorithms and training
torchelie.nn.utils.
remove_batchnorm
Remove BatchNorm in Sequentials / CondSeqs in a smart way, restoring biases in the preceding layer.