An implementation of WideResNets with Fixup initialization in Jax/Flax. This can be useful for use cases where Batch Normalization should be avoided (for example when using the Laplace approximation to the Bayesian posterior).
-
Updated
Jul 19, 2023 - Python