antsnormflows.flows.normalization

Classes

ActNorm(*args[, log_s_cap])

Glow-style ActNorm with data-dependent init.

BatchNorm([eps, detach_stats])

RealNVP-style BatchNorm that ignores gradients through batch stats.