def init_weights(module, weight_mean=0, weight_stddev=0.1, bias_mean=0.1):
    """ Init weights of torch.module. """
    module.weight.data.normal_(weight_mean, weight_stddev)
    module.bias.data.fill_(bias_mean)
    return module