mednet.engine.uploader

Functions to upload models and measures to GitLab’s experiment registry.

Functions

run(project_path, experiment_folder, ...)

Upload results from an experiment folder to GitLab's MLFlow server.

mednet.engine.uploader.run(project_path, experiment_folder, experiment_name, run_name, metrics, upload_limit_mb)[source]

Upload results from an experiment folder to GitLab’s MLFlow server.

Parameters:
  • project_path (str) – Path to the project where to upload model entries.

  • experiment_folder (Path) – Directory in which to upload results from.

  • experiment_name (str) – A string indicating the experiment name (e.g. “exp-pasa-mc” or “exp-densenet-mc-ch”).

  • run_name (str) – A string indicating the run name (e.g. “run-1”).

  • metrics (list[str]) – List of metrics to upload.

  • upload_limit_mb (int) – Maximim upload size in MB (set to 0 for no limit).

Return type:

None