mlflow for Auto3Dseg

[YazdanSalimi]

Hello.
I have a problem with running auto3dseg two minutes example.
the issue is with mlflow I tried multiple solution online, non of them worked. Could you please help to resolve it?
the error message is:
2024-05-14 18:58:35,964 - INFO - AutoRunner using work directory C:\auto3dsegexample
2024-05-14 18:58:35,966 - INFO - Found num_fold 3 based on the input datalist C:\auto3dsegexample\sim_datalist.json.
2024-05-14 18:58:35,966 - INFO - Setting num_fold 3 based on the input datalist C:\auto3dsegexample\sim_datalist.json.
2024-05-14 18:58:35,992 - INFO - Using user defined command running prefix , will override other settings
2024-05-14 18:58:35,992 - INFO - Running data analysis...
2024-05-14 18:58:35,993 - INFO - Found 1 GPUs for data analyzing!
0%| | 0/12 [00:00<?, ?it/s]2024-05-14 18:58:39,693 - INFO - AutoRunner using work directory C:\auto3dsegexample
2024-05-14 18:58:39,695 - INFO - AutoRunner using work directory C:\auto3dsegexample
2024-05-14 18:58:39,696 - INFO - Found num_fold 3 based on the input datalist C:\auto3dsegexample\sim_datalist.json.
2024-05-14 18:58:39,696 - INFO - Setting num_fold 3 based on the input datalist C:\auto3dsegexample\sim_datalist.json.
2024-05-14 18:58:39,697 - INFO - Found num_fold 3 based on the input datalist C:\auto3dsegexample\sim_datalist.json.
2024-05-14 18:58:39,697 - INFO - Setting num_fold 3 based on the input datalist C:\auto3dsegexample\sim_datalist.json.
2024-05-14 18:58:39,721 - INFO - AutoRunner using work directory C:\auto3dsegexample
2024-05-14 18:58:39,723 - INFO - Found num_fold 3 based on the input datalist C:\auto3dsegexample\sim_datalist.json.
2024-05-14 18:58:39,723 - INFO - Setting num_fold 3 based on the input datalist C:\auto3dsegexample\sim_datalist.json.
2024-05-14 18:58:39,731 - INFO - Using user defined command running prefix , will override other settings
2024-05-14 18:58:39,745 - INFO - Using user defined command running prefix , will override other settings
2024-05-14 18:58:39,761 - INFO - Using user defined command running prefix , will override other settings
2024-05-14 18:58:39,790 - INFO - AutoRunner using work directory C:\auto3dsegexample
2024-05-14 18:58:39,797 - INFO - Found num_fold 3 based on the input datalist C:\auto3dsegexample\sim_datalist.json.
2024-05-14 18:58:39,797 - INFO - Setting num_fold 3 based on the input datalist C:\auto3dsegexample\sim_datalist.json.
2024-05-14 18:58:39,898 - INFO - Using user defined command running prefix , will override other settings
100%|██████████████████████████████████████████████████████████████████████████████████| 12/12 [00:05<00:00, 2.18it/s]
2024-05-14 18:58:41,546 - INFO - Writing data stats to C:\auto3dsegexample\datastats.yaml.
2024-05-14 18:58:41,555 - INFO - Writing by-case data stats to C:\auto3dsegexample\datastats_by_case.yaml, this may take a while.
2024-05-14 18:58:41,610 - INFO - BundleGen from https://github.com/Project-MONAI/research-contributions/releases/download/algo_templates/e4cf5a1.tar.gz
algo_templates.tar.gz: 104kB [00:01, 105kB/s]
2024-05-14 18:58:42,626 - INFO - Downloaded: C:\Users\Yazdan\AppData\Local\Temp\tmpur3hocyk\algo_templates.tar.gz
2024-05-14 18:58:42,626 - INFO - Expected md5 is None, skip md5 check for file C:\Users\Yazdan\AppData\Local\Temp\tmpur3hocyk\algo_templates.tar.gz.
2024-05-14 18:58:42,626 - INFO - Writing into directory: C:\auto3dsegexample.
2024-05-14 18:58:42,902 - INFO - Generated:C:\auto3dsegexample\dints_0
2024-05-14 18:58:42,963 - INFO - Generated:C:\auto3dsegexample\segresnet_0
2024-05-14 18:58:42,990 - INFO - segresnet2d_0 is skipped! SegresNet2D is skipped due to median spacing of [1.0, 1.0, 1.0], which means the dataset is not highly anisotropic, e.g. spacing[2] < 3*(spacing[0] + spacing[1])/2) .
2024-05-14 18:58:43,067 - INFO - Generated:C:\auto3dsegexample\swinunetr_0
2024-05-14 18:58:43,118 - INFO - The keys num_warmup_epochs cannot be found in the C:\auto3dsegexample\dints_0\configs\hyper_parameters.yaml for training. Skipped overriding key num_warmup_epochs.
2024-05-14 18:58:43,119 - INFO - ['python', 'C:\auto3dsegexample\dints_0\scripts\train.py', 'run', "--config_file='C:/auto3dsegexample/dints_0/configs/hyper_parameters.yaml,C:/auto3dsegexample/dints_0/configs/hyper_parameters_search.yaml,C:/auto3dsegexample/dints_0/configs/network.yaml,C:/auto3dsegexample/dints_0/configs/network_search.yaml,C:/auto3dsegexample/dints_0/configs/transforms_infer.yaml,C:/auto3dsegexample/dints_0/configs/transforms_train.yaml,C:/auto3dsegexample/dints_0/configs/transforms_validate.yaml'", '--training#num_epochs_per_validation=1', '--training#num_images_per_batch=2', '--training#num_epochs=2']
Warning: cannot find libcudart.so or AMD ROCm platform, set device limit is disabled
Traceback (most recent call last):
File "C:\Users\Yazdan\anaconda3\envs\nn\lib\site-packages\mlflow\tracking\registry.py", line 81, in get_store_builder
store_builder = self._registry[scheme]
KeyError: 'c'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "C:\auto3dsegexample\dints_0\scripts\train.py", line 1002, in
fire.Fire()
File "C:\Users\Yazdan\anaconda3\envs\nn\lib\site-packages\fire\core.py", line 143, in Fire
component_trace = _Fire(component, args, parsed_flag_args, context, name)
File "C:\Users\Yazdan\anaconda3\envs\nn\lib\site-packages\fire\core.py", line 477, in _Fire
component, remaining_args = _CallAndUpdateTrace(
File "C:\Users\Yazdan\anaconda3\envs\nn\lib\site-packages\fire\core.py", line 693, in _CallAndUpdateTrace
component = fn(*varargs, **kwargs)
File "C:\auto3dsegexample\dints_0\scripts\train.py", line 535, in run
mlflow.set_experiment(mlflow_experiment_name)
File "C:\Users\Yazdan\anaconda3\envs\nn\lib\site-packages\mlflow\tracking\fluent.py", line 141, in set_experiment
client = MlflowClient()
File "C:\Users\Yazdan\anaconda3\envs\nn\lib\site-packages\mlflow\tracking\client.py", line 100, in init
self._tracking_client = TrackingServiceClient(final_tracking_uri)
File "C:\Users\Yazdan\anaconda3\envs\nn\lib\site-packages\mlflow\tracking_tracking_service\client.py", line 50, in init
self.store
File "C:\Users\Yazdan\anaconda3\envs\nn\lib\site-packages\mlflow\tracking_tracking_service\client.py", line 54, in store
return utils._get_store(self.tracking_uri)
File "C:\Users\Yazdan\anaconda3\envs\nn\lib\site-packages\mlflow\tracking_tracking_service\utils.py", line 197, in _get_store
return _tracking_store_registry.get_store(store_uri, artifact_uri)
File "C:\Users\Yazdan\anaconda3\envs\nn\lib\site-packages\mlflow\tracking_tracking_service\registry.py", line 42, in get_store
return self._get_store_with_resolved_uri(resolved_store_uri, artifact_uri)
File "C:\Users\Yazdan\anaconda3\envs\nn\lib\site-packages\mlflow\tracking_tracking_service\registry.py", line 51, in _get_store_with_resolved_uri
builder = self.get_store_builder(resolved_store_uri)
File "C:\Users\Yazdan\anaconda3\envs\nn\lib\site-packages\mlflow\tracking\registry.py", line 83, in get_store_builder
raise UnsupportedModelRegistryStoreURIException(
mlflow.tracking.registry.UnsupportedModelRegistryStoreURIException: Model registry functionality is unavailable; got unsupported URI 'C:\auto3dsegexample\dints_0/model_fold0/mlruns/' for model registry data storage. Supported URI schemes are: ['', 'file', 'databricks', 'databricks-uc', 'http', 'https', 'postgresql', 'mysql', 'sqlite', 'mssql']. See https://www.mlflow.org/docs/latest/tracking.html#storage for how to run an MLflow server against one of the supported backend storage locations.
Traceback (most recent call last):
File "C:\Users\Yazdan\Downloads\auto3dseg.py", line 86, in
runner.run()
File "C:\Users\Yazdan\anaconda3\envs\nn\lib\site-packages\monai\apps\auto3dseg\auto_runner.py", line 878, in run
self._train_algo_in_sequence(history)
File "C:\Users\Yazdan\anaconda3\envs\nn\lib\site-packages\monai\apps\auto3dseg\auto_runner.py", line 728, in _train_algo_in_sequence
algo.train(self.train_params, self.device_setting)
File "C:\auto3dsegexample\algorithm_templates\dints\scripts\algo.py", line 496, in train
return self._run_cmd(cmd, devices_info)
File "C:\Users\Yazdan\anaconda3\envs\nn\lib\site-packages\monai\apps\auto3dseg\bundle_gen.py", line 277, in _run_cmd
return run_cmd(cmd.split(), run_cmd_verbose=True, env=ps_environ, check=True)
File "C:\Users\Yazdan\anaconda3\envs\nn\lib\site-packages\monai\utils\misc.py", line 873, in run_cmd
return subprocess.run(cmd_list, **kwargs)
File "C:\Users\Yazdan\anaconda3\envs\nn\lib\subprocess.py", line 528, in run
raise CalledProcessError(retcode, process.args,
subprocess.CalledProcessError: Command '['python', 'C:\auto3dsegexample\dints_0\scripts\train.py', 'run', "--config_file='C:/auto3dsegexample/dints_0/configs/hyper_parameters.yaml,C:/auto3dsegexample/dints_0/configs/hyper_parameters_search.yaml,C:/auto3dsegexample/dints_0/configs/network.yaml,C:/auto3dsegexample/dints_0/configs/network_search.yaml,C:/auto3dsegexample/dints_0/configs/transforms_infer.yaml,C:/auto3dsegexample/dints_0/configs/transforms_train.yaml,C:/auto3dsegexample/dints_0/configs/transforms_validate.yaml'", '--training#num_epochs_per_validation=1', '--training#num_images_per_batch=2', '--training#num_epochs=2']' returned non-zero exit status 1.
Thank you

[KumoLiu]

Hi @YazdanSalimi, thanks for your interest here. The current version of Auto3DSeg algorithm use a Linux library and can't be used on Windows at this point.
Project-MONAI/MONAI#6821 (comment)