Open dlwh opened 1 year ago
See our LayerNorm and Eqx's groupnorm https://docs.kidger.site/equinox/api/nn/normalisation/#equinox.nn.GroupNorm
See our LayerNorm and Eqx's groupnorm https://docs.kidger.site/equinox/api/nn/normalisation/#equinox.nn.GroupNorm