Weights are not updated

See also Hessian inside a Flux loss function