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)