def __init__(self, preprocess_net, hidden_layer_size=128):
    super().__init__()
    self.preprocess = preprocess_net
    self.last = nn.Linear(hidden_layer_size, 1)
def __init__(self, preprocess_net, hidden_layer_size=128):
    super().__init__()
    self.preprocess = preprocess_net
    self.last = nn.Linear(hidden_layer_size, 1)