def __init__()

in scripts/train.py [0:0]


    def __init__(self, x1_size, x2_size):
        super(Net, self).__init__()
        self.batch_norm = nn.BatchNorm1d(x1_size)
        self.fc1 = nn.Linear(x1_size, 10)
        self.fc2 = nn.Linear(10 + x2_size, 10)
        self.fc3 = nn.Linear(10, 1)