MLFlowHandler.__init__() got an unexpected keyword argument 'experiment_name'

[tarek909]

Hello,

I'm using monai radiology app, and segmentation_spleen task

When I connect with 3D slicer, run model, correct annotation of the data, submit data all of these steps goes well without any problems, but when I click on train to start training the model, the progress bar counts to 100% but the status change from "1/1 epochs completed" to "ERROR" and this is my logs, any help will be appreciated.

[2023-04-09 02:09:42,628] [20348] [MainThread] [INFO] (main:126) - +++ Adding Inferer:: segmentation_spleen => <lib.infers.segmentation_spleen.SegmentationSpleen object at 0x0000019B148C7610>
[2023-04-09 02:09:42,629] [20348] [MainThread] [INFO] (main:191) - {'segmentation_spleen': <lib.infers.segmentation_spleen.SegmentationSpleen object at 0x0000019B148C7610>, 'Histogram+GraphCut': <monailabel.scribbles.infer.HistogramBasedGraphCut object at 0x0000019B1492EB60>, 'GMM+GraphCut': <monailabel.scribbles.infer.GMMBasedGraphCut object at 0x0000019B1492EB30>}
[2023-04-09 02:09:42,629] [20348] [MainThread] [INFO] (main:206) - +++ Adding Trainer:: segmentation_spleen => <lib.trainers.segmentation_spleen.SegmentationSpleen object at 0x0000019B1492EE00>
[2023-04-09 02:09:42,629] [20348] [MainThread] [INFO] (monailabel.utils.sessions:51) - Session Path: C:\Users\Tarek.cache\monailabel\sessions
[2023-04-09 02:09:42,629] [20348] [MainThread] [INFO] (monailabel.utils.sessions:52) - Session Expiry (max): 3600
[2023-04-09 02:09:42,629] [20348] [MainThread] [INFO] (monailabel.tasks.train.basic_train:432) - Train Request (input): {'model': 'segmentation_spleen', 'name': 'train_01', 'pretrained': True, 'device': 'cuda', 'max_epochs': 50, 'early_stop_patience': -1, 'val_split': 0.2, 'train_batch_size': 1, 'val_batch_size': 1, 'multi_gpu': True, 'gpus': 'all', 'dataset': 'SmartCacheDataset', 'dataloader': 'ThreadDataLoader', 'tracking': 'mlflow', 'tracking_uri': '', 'tracking_experiment_name': '', 'client_id': 'user-xyz', 'local_rank': 0}
[2023-04-09 02:09:42,629] [20348] [MainThread] [INFO] (monailabel.tasks.train.basic_train:442) - CUDA_VISIBLE_DEVICES: None
[2023-04-09 02:09:42,630] [20348] [MainThread] [INFO] (monailabel.tasks.train.basic_train:447) - Distributed/Multi GPU is limited
[2023-04-09 02:09:42,630] [20348] [MainThread] [INFO] (monailabel.tasks.train.basic_train:462) - Distributed Training = FALSE
[2023-04-09 02:09:42,630] [20348] [MainThread] [INFO] (monailabel.tasks.train.basic_train:489) - 0 - Train Request (final): {'name': 'train_01', 'pretrained': True, 'device': 'cuda', 'max_epochs': 50, 'early_stop_patience': -1, 'val_split': 0.2, 'train_batch_size': 1, 'val_batch_size': 1, 'multi_gpu': False, 'gpus': 'all', 'dataset': 'SmartCacheDataset', 'dataloader': 'ThreadDataLoader', 'tracking': 'mlflow', 'tracking_uri': '', 'tracking_experiment_name': '', 'model': 'segmentation_spleen', 'client_id': 'user-xyz', 'local_rank': 0, 'run_id': '20230409_020942'}
[2023-04-09 02:09:42,630] [20348] [MainThread] [INFO] (monailabel.tasks.train.basic_train:622) - 0 - Using Device: cpu; IDX: None
[2023-04-09 02:09:42,631] [20348] [MainThread] [INFO] (monailabel.tasks.train.basic_train:515) - Run/Output Path: D:\GP\Spleen\test\workspace\radiology\model\segmentation_spleen\train_01
[2023-04-09 02:09:42,631] [20348] [MainThread] [INFO] (monailabel.tasks.train.basic_train:531) - Tracking: mlflow
[2023-04-09 02:09:42,631] [20348] [MainThread] [INFO] (monailabel.tasks.train.basic_train:532) - Tracking URI: file:///D:/GP/Spleen/test/workspace/radiology/model/segmentation_spleen/train_01/mlruns;
[2023-04-09 02:09:42,631] [20348] [MainThread] [INFO] (monailabel.tasks.train.basic_train:533) - Tracking Experiment Name: segmentation_spleen; Run Name: run_20230409_020942
[2023-04-09 02:09:42,631] [20348] [MainThread] [INFO] (monailabel.tasks.train.basic_train:410) - Total Records for Training: 1
[2023-04-09 02:09:42,631] [20348] [MainThread] [INFO] (monailabel.tasks.train.basic_train:411) - Total Records for Validation: 0
Traceback (most recent call last):
File "C:\Users\Tarek\AppData\Local\Programs\Python\Python310\lib\runpy.py", line 196, in _run_module_as_main
return _run_code(code, main_globals, None,
File "C:\Users\Tarek\AppData\Local\Programs\Python\Python310\lib\runpy.py", line 86, in _run_code
exec(code, run_globals)
File "C:\Users\Tarek\AppData\Local\Programs\Python\Python310\lib\site-packages\monailabel\interfaces\utils\app.py", line 128, in
run_main()
File "C:\Users\Tarek\AppData\Local\Programs\Python\Python310\lib\site-packages\monailabel\interfaces\utils\app.py", line 113, in run_main
result = a.train(request)
File "C:\Users\Tarek\AppData\Local\Programs\Python\Python310\lib\site-packages\monailabel\interfaces\app.py", line 422, in train
result = task(request, self.datastore())
File "C:\Users\Tarek\AppData\Local\Programs\Python\Python310\lib\site-packages\monailabel\tasks\train\basic_train.py", line 463, in call
res = self.train(0, world_size, req, datalist)
File "C:\Users\Tarek\AppData\Local\Programs\Python\Python310\lib\site-packages\monailabel\tasks\train\basic_train.py", line 541, in train
context.trainer = self._create_trainer(context)
File "C:\Users\Tarek\AppData\Local\Programs\Python\Python310\lib\site-packages\monailabel\tasks\train\basic_train.py", line 671, in _create_trainer
train_handlers: List = self.train_handlers(context)
File "C:\Users\Tarek\AppData\Local\Programs\Python\Python310\lib\site-packages\monailabel\tasks\train\basic_train.py", line 307, in train_handlers
MLFlowHandler(
TypeError: MLFlowHandler.init() got an unexpected keyword argument 'experiment_name'
[2023-04-09 02:09:43,474] [16732] [ThreadPoolExecutor-2_0] [INFO] (monailabel.utils.async_tasks.utils:83) - Return code: 1

[diazandr3s]

HI @tarek909,

Thanks for opening this discussion.
It seems you're using an old version of the Radiology app. Please use the latest MONAI Label version and the latest Radiology app. A clean installation will help.

Let us know how that goes

[tarek909]

It works now! Thanks a lot for your help