`evorl.networks.layer_norm`¶

Module Contents¶

Classes¶

StaticLayerNorm

Layer normalization with fixed scale and bias.

Functions¶

get_norm_layer

Get the normalization layer class based on the type.

API¶

class evorl.networks.layer_norm.StaticLayerNorm[source]¶

Bases: flax.linen.LayerNorm

Layer normalization with fixed scale and bias.

fixed_bias: jax.Array¶: ‘field(…)’

fixed_scale: jax.Array¶: ‘field(…)’

use_bias: bool¶: False

use_scale: bool¶: False

evorl.networks.layer_norm.get_norm_layer(norm_layer_type: str) → type[flax.linen.Module][source]¶: Get the normalization layer class based on the type.