mirror of
https://github.com/wassname/ray.git
synced 2026-09-12 12:51:15 +08:00
[Tune] Mlflow Integration (#12840)
Co-authored-by: Kai Fricke <krfricke@users.noreply.github.com> Co-authored-by: Richard Liaw <rliaw@berkeley.edu>
This commit is contained in:
co-authored by
Kai Fricke
Richard Liaw
parent
5d987f5988
commit
5d3c9c8861
@@ -70,11 +70,7 @@ An example of creating a custom logger can be found in :doc:`/tune/examples/logg
|
||||
Trainable Logging
|
||||
-----------------
|
||||
|
||||
By default, Tune only logs the *training result dictionaries* from your Trainable. However, you may want to visualize the model weights, model graph, or use a custom logging library that requires multi-process logging. For example, you may want to do this if:
|
||||
|
||||
* you're using `Weights and Biases <https://www.wandb.com/>`_
|
||||
* you're using `MLFlow <https://github.com/mlflow/mlflow/>`__
|
||||
* you're trying to log images to Tensorboard.
|
||||
By default, Tune only logs the *training result dictionaries* from your Trainable. However, you may want to visualize the model weights, model graph, or use a custom logging library that requires multi-process logging. For example, you may want to do this if you're trying to log images to Tensorboard.
|
||||
|
||||
You can do this in the trainable, as shown below:
|
||||
|
||||
@@ -163,12 +159,17 @@ CSVLogger
|
||||
|
||||
.. autoclass:: ray.tune.logger.CSVLoggerCallback
|
||||
|
||||
MLFLowLogger
|
||||
MLFlowLogger
|
||||
------------
|
||||
|
||||
Tune also provides a default logger for `MLFlow <https://mlflow.org>`_. You can install MLFlow via ``pip install mlflow``. An example can be found in :doc:`/tune/examples/mlflow_example`. Note that this currently does not include artifact logging support. For this, you can use the native MLFlow APIs inside your Trainable definition.
|
||||
Tune also provides a default logger for `MLFlow <https://mlflow.org>`_. You can install MLFlow via ``pip install mlflow``.
|
||||
You can see the :doc:`tutorial here </tune/tutorials/tune-mlflow>`.
|
||||
|
||||
.. autoclass:: ray.tune.logger.MLFLowLogger
|
||||
WandbLogger
|
||||
-----------
|
||||
|
||||
Tune also provides a default logger for `Weights & Biases <https://www.wandb.com/>`_. You can install Wandb via ``pip install wandb``.
|
||||
You can see the :doc:`tutorial here </tune/tutorials/tune-wandb>`
|
||||
|
||||
|
||||
.. _logger-interface:
|
||||
|
||||
Reference in New Issue
Block a user