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

Hyperparameter optimization with Optuna #24

Merged
merged 11 commits into from
Sep 20, 2024
Merged

Commits on Sep 11, 2024

  1. Main file for hyperparameter tuning:

    1. Running with a .yaml config file will lead to usual execution.
    2. Running with the --optimized flag will tune hyperparameters with optuna. No input file needed.
    divijghose committed Sep 11, 2024
    Configuration menu
    Copy the full SHA
    27041fb View commit details
    Browse the repository at this point in the history
  2. Module for hyperparameter tuning.

    1. objective.py defines the objective function for tuning. Contains the fastvpinns object returning metric for tuning.
    2. optuna_tuner.py manages the hyperparameter tuning process.
    divijghose committed Sep 11, 2024
    Configuration menu
    Copy the full SHA
    65637b7 View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    9ac45cc View commit details
    Browse the repository at this point in the history
  4. Configuration menu
    Copy the full SHA
    5aff902 View commit details
    Browse the repository at this point in the history

Commits on Sep 18, 2024

  1. Configuration menu
    Copy the full SHA
    9d1618b View commit details
    Browse the repository at this point in the history
  2. Changes to main file to incorporate hyperparameter tuning using Optuna

    1. Accept number of trials and number of training iteration for each trial as an argument.
    divijghose committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    dc1a7f9 View commit details
    Browse the repository at this point in the history
  3. Changes to geometry fle:

    1. Accept an is_optimized argument, True if hyperparameter optimization with Optuna is being used.
    2. If is_optimized is True, geometry module doesn't print out the test mesh and VTK file for each trial.
    3. Backward compatibility - default value of is_optimized is False, existing code with config file should work as is.
    divijghose committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    3dd7f00 View commit details
    Browse the repository at this point in the history
  4. Parallel runs with optuna tuner

    1. Creates an SQLite database if it doesn't exist. Can be used for stalled runs or parallel implementation.
    2. Lists available number of GPUs and divides jobs.
    divijghose committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    6e05813 View commit details
    Browse the repository at this point in the history
  5. Configuration menu
    Copy the full SHA
    3b63a4c View commit details
    Browse the repository at this point in the history
  6. Configuration menu
    Copy the full SHA
    7aa1711 View commit details
    Browse the repository at this point in the history
  7. Black formatting.

    divijghose committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    d09e32f View commit details
    Browse the repository at this point in the history