Function reset_parameters

lit_gpt/utils.py:60–64

Calls `reset_parameters` on the module and all its submodules.

(module: nn.Module)

def reset_parameters(module: nn.Module) -> None:
    """Calls `reset_parameters` on the module and all its submodules."""
    for mod in module.modules():
        if callable(getattr(mod, 'reset_parameters', None)):
            mod.reset_parameters()


def check_valid_checkpoint_dir(