Changed convention to match Fyodorov.

1 files changed, 2 insertions, 2 deletions

diff --git a/least_squares.cpp b/least_squares.cpp
index 28397ec..f667e28 100644
--- a/least_squares.cpp
+++ b/least_squares.cpp
@@ -97,7 +97,7 @@ public:
   std::tuple<Real, Vector, Matrix> hamGradHess(const Vector& x) const {
     auto [H, dH, ddH] = HdHddH(x);
 
-    Vector gradH = dH - dH.dot(x) * x / N;
+    Vector gradH = dH - dH.dot(x) * x / (Real)N;
     Matrix hessH = ddH - (dH * x.transpose() + x.dot(dH) * Matrix::Identity(N, N) + (ddH * x) * x.transpose()) / (Real)N  + 2.0 * x * x.transpose();
 
     return {H, gradH, hessH};
@@ -171,7 +171,7 @@ int main(int argc, char* argv[]) {
 
   Rng r;
 
-  Model<1, 2> leastSquares(N, M, r.engine(), sqrt(2) * pow(σ, 2), sqrt(2));
+  Model<1, 2> leastSquares(N, M, r.engine(), σ, 1);
 
   Vector x = Vector::Zero(N);
   x(0) = sqrt(N);