def trunc_normal_init(module: nn.Module, mean: float=0, std: float=1, a: float=(- 2), b: float=2, bias: float=0) -> None:
    if (hasattr(module, 'weight') and (module.weight is not None)):
        trunc_normal_(module.weight, mean, std, a, b)
    if (hasattr(module, 'bias') and (module.bias is not None)):
        nn.init.constant_(module.bias, bias)
def trunc_normal_init(module: nn.Module, mean: float=0, std: float=1, a: float=(- 2), b: float=2, bias: float=0) -> None:
    if (hasattr(module, 'weight') and (module.weight is not None)):
        trunc_normal_(module.weight, mean, std, a, b)
    if (hasattr(module, 'bias') and (module.bias is not None)):
        nn.init.constant_(module.bias, bias)