Function set_module_eval_mode

src/sharp/utils/module_surgery.py:22–23

(module: nn.Module, _: Any)

    """Freeze all normalization layers."""

    def set_module_eval_mode(module: nn.Module, _: Any) -> None:
        module.eval()

    for submodule in module.modules():
        if isinstance(submodule, NORM_LAYER_TYPES):