Logging model checkpoints as artifacts in MlFlow

[AlirezaMorsali]

What does this PR do ?

MlFlow is required for training and resuming runs from checkpoint. Currently, when the MlFlow logger is enabled in experiment manager it doesn't log model checkpoint as artifacts to MLFlow.
This PR simply adds one line to the nemo.utils.loggers.MLFlowParams to allow for logging checkpoints.

Collection: [Note which collection this PR will affect]

Changelog

• Add log_model to nemo.utils.loggers.MLFlowParams

Usage

To enable logging checkpoints in MlFlow

exp_manager:
  ...
  create_checkpoint_callback: True
  create_mlflow_logger: True
  mlflow_logger_kwargs:
  ...
      log_model: True

Before your PR is "Ready for review"

Pre checks:

• Make sure you read and followed Contributor guidelines
• Did you write any new necessary tests?
• Did you add or update any necessary documentation?
• Does the PR affect components that are optional to install? (Ex: Numba, Pynini, Apex etc)
  • Reviewer: Does the PR have correct import guards for all optional libraries?

PR Type:

• New Feature
• Bugfix
• Documentation

If you haven't finished some of the above items you can still open "Draft" PR.

Who can review?

Anyone in the community is free to review the PR once the checks have passed.
Contributor guidelines contains specific people who can review PRs to various areas.

Additional Information

• Related to Experiment manager MlFlow logger does not log model checkpoints as artifacts #7257

[titu1994]

Thanks !