def METHOD_NAME(self, module):
    if isinstance(module, torch.nn.Linear):
        module.weight.data.normal_(mean=0.0, std=0.02)
    if isinstance(module, torch.nn.Linear) and module.bias is not None:
        module.bias.data.zero_()