def weights_init_normal()

in torchrecipes/vision/image_generation/module/infogan.py [0:0]

• 7 lines of code
• 6 McCabe index (conditional complexity)

def weights_init_normal(m: nn.Module) -> None:
    classname = m.__class__.__name__
    if classname.find("Conv") != -1:
        # pyre-fixme[6]: Expected `Tensor` for 1st positional only
        # parameter to call `nn.init.normal_` but got
        # `typing.Union[Tensor, nn.Module]`
        torch.nn.init.normal_(m.weight.data, 0.0, 0.02)
    elif classname.find("BatchNorm") != -1:
        # pyre-fixme[6]: Expected `Tensor` for 1st positional only
        # parameter to call `nn.init.normal_` but got
        # `typing.Union[Tensor, nn.Module]`
        torch.nn.init.normal_(m.weight.data, 1.0, 0.02)
        # pyre-fixme[6]: Expected `Tensor` for 1st positional only
        # parameter to call `nn.init.constant_` but got
        # `typing.Union[Tensor, nn.Module]`
        torch.nn.init.constant_(m.bias.data, 0.0)