Function num_parameters

lit_gpt/utils_old.py:27–28

(module: nn.Module, requires_grad: Optional[bool] = None)

def num_parameters(module: nn.Module, requires_grad: Optional[bool] = None) -> int:
    return sum(p.numel() for p in module.parameters() if requires_grad is None or p.requires_grad == requires_grad)


@contextmanager