def __init__(self, init_weight=4):
        """Initialization of network and filling it with given numbers."""
        super(Net, self).__init__()
        self.fc = nn.Linear(2, 1)
        self.fc.weight.data.fill_(init_weight)
        self.fc.bias.data.fill_(init_weight)
Initialization of network and filling it with given numbers.
tests/contrib/utils/test_swa.py