Skip to content

GitLab

  • Menu
Projects Groups Snippets
    • Loading...
  • Help
    • Help
    • Support
    • Community forum
    • Submit feedback
    • Contribute to GitLab
  • Sign in
  • A AIPassport
  • Project information
    • Project information
    • Activity
    • Labels
    • Planning hierarchy
    • Members
  • Repository
    • Repository
    • Files
    • Commits
    • Branches
    • Tags
    • Contributors
    • Graph
    • Compare
  • Issues 0
    • Issues 0
    • List
    • Boards
    • Service Desk
    • Milestones
  • Merge requests 2
    • Merge requests 2
  • CI/CD
    • CI/CD
    • Pipelines
    • Jobs
    • Schedules
  • Deployments
    • Deployments
    • Environments
    • Releases
  • Monitor
    • Monitor
    • Metrics
    • Incidents
  • Packages & Registries
    • Packages & Registries
    • Package Registry
    • Infrastructure Registry
  • Analytics
    • Analytics
    • Value stream
    • CI/CD
    • Repository
  • Wiki
    • Wiki
  • Snippets
    • Snippets
  • Activity
  • Graph
  • Create a new issue
  • Jobs
  • Commits
  • Issue Boards
Collapse sidebar
  • Charalampos Kalantzopoulos
  • AIPassport
  • Merge requests
  • !3

Open
Created Apr 09, 2025 by Ioannis Kyprakis@ikyprakisMaintainer
  • Report abuse
Report abuse

Hyperparameter Search Feature

  • Overview 0
  • Commits 1
  • Changes 11

When a list or a string starting with "range" is found in the params of a stage, Passport.AddStage() creates a DVC stage using -foreach and -do, passing each combination of those hyperparameters as an argument.

In the MLflowTracker, when such list or string values are detected in the stage's params, a nested run is created under the name of the stage, with child runs named stage_name_{run_number}. DS_VERSION is updated to include the run IDs of the child runs. Each run's hyperparameter set is logged to MLflow.

Additionally, when torch is found in sys.modules, mlflow.pytorch.autolog() is used instead of mlflow.autolog().

In return_md5, if foreach is present in the stage, the cmd is under 'do', and is parsed accordingly.

Assignee
Assign to
Reviewer
Request review from
Time tracking
Source branch: HP_search_feature