diff --git a/neural_network_lyapunov/test/train_toy_system_controller_demo.py b/neural_network_lyapunov/test/train_toy_system_controller_demo.py index c58aa3bf..b504aa81 100644 --- a/neural_network_lyapunov/test/train_toy_system_controller_demo.py +++ b/neural_network_lyapunov/test/train_toy_system_controller_demo.py @@ -82,7 +82,7 @@ dut = train_lyapunov_barrier.Trainer() dut.add_lyapunov( lyapunov_hybrid_system, V_lambda, x_equilibrium, - r_options.FixedROptions(torch.eye(2, dtype=torch.float64))) + r_options.FixedROptions(torch.eye(args.dimension, dtype=torch.float64))) dut.lyapunov_positivity_mip_pool_solutions = 1 dut.lyapunov_derivative_mip_pool_solutions = 1 dut.lyapunov_derivative_convergence_tol = 5E-5