Is your feature request related to a problem? Please describe.
Describe the solution you'd like
MLFlowHandler is being used to log DL experiments with PyTorch Ignite for MONAI bundles. We can further enhance the module by adding these four features:
P0:
- API for users to add experiment/run name in MLFlow
- API for users to log customized params for each run
- Methods to log json file content
- Methods to log result images
- Methods to log optimizer params
- Make sure this handler works in multi-gpu & multi-node environment
- Make sure this handler works in all existed bundles
P1:
- Methods to log dataset
- Enhance the
tracking_uri parameter for convenient modification
Describe alternatives you've considered
Use ignite.contrib.handlers.mlflow_logger
https://pytorch.org/ignite/generated/ignite.contrib.handlers.mlflow_logger.html#module-ignite.contrib.handlers.mlflow_logger
Is your feature request related to a problem? Please describe.
Describe the solution you'd like
MLFlowHandler is being used to log DL experiments with PyTorch Ignite for MONAI bundles. We can further enhance the module by adding these four features:
P0:
P1:
tracking_uriparameter for convenient modificationDescribe alternatives you've considered
Use
ignite.contrib.handlers.mlflow_loggerhttps://pytorch.org/ignite/generated/ignite.contrib.handlers.mlflow_logger.html#module-ignite.contrib.handlers.mlflow_logger