fit() print(loss_func(model(xb), yb))