def METHOD_NAME(self, targets, outputs, loss_fn, inputs, model, aux=None):
    n = len(self.X_lo_train)
    loss_lo = loss_fn(targets[0][:n], outputs[0][:n])
    loss_hi = loss_fn(targets[1][n:], outputs[1][n:])
    return [loss_lo, loss_hi]