Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

verbose=False is not working as expected for NonlinearCG #571

Closed
Joshuaalbert opened this issue Jan 23, 2024 · 1 comment · Fixed by #573
Closed

verbose=False is not working as expected for NonlinearCG #571

Joshuaalbert opened this issue Jan 23, 2024 · 1 comment · Fixed by #573

Comments

@Joshuaalbert
Copy link

    def loss(U: UType):
        return -model.log_prob_likelihood(U, allow_nan=False)

    solver = NonlinearCG(
        fun=loss,
        jit=True,
        unroll=False,
        verbose=False
    )
    results = solver.run(init_params=init_U_point)

It still prints out linesearch debug lines like:

WARNING: jaxopt.ZoomLineSearch: Returning stepsize with sufficient decrease but curvature condition not satisfied.
INFO: jaxopt.ZoomLineSearch: Iter: 9 Minimum Decrease & Curvature Errors (stop. crit.): 3.3350500139306405e-09 Stepsize:0.00016820144082885236  Decrease Error:0.0  Curvature Error:3.3350500139306405e-09 
@Joshuaalbert Joshuaalbert changed the title verbose=False is not working as expected verbose=False is not working as expected for NonlinearCG Jan 24, 2024
@vroulet
Copy link
Collaborator

vroulet commented Jan 25, 2024

Thanks for bringing this up. #573 should fix the issue, but I need to investigate the failing tests.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging a pull request may close this issue.

2 participants