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

CLI format run benchopt #2

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

tomMoral
Copy link

A first POC to run this benchmark with benchopt.

The results looks like these after running benchopt run benchopt_benchmark/ -n 100 -s SGD[max_epochs=20,learning_rate=[2e-3,1e-3]] -d MNIST[seed=42,activation=ReLU] -r 2
image

(note that you probably need to install the dev version of benchopt as pip install git+https://github.com/benchopt/benchopt to have the plot running.)

There are several issues with this implem:

  • There are no checkpoints yet.
  • The test loss corresponds to the last and not the best
  • The train loss is incorrect as I did not know how to access it from the Trainer.
  • There are several things that I changed directly in FOB to avoid having to mess too much with the config file. We could do either depending on what you thing make more sense.

But that is already a good basis to discuss if this makes sense and if we should continue further :)
Happy to chat IRL in ICML if you are going!

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 this pull request may close these issues.

1 participant