2020-08-31 23:30:48 +01:00
|
|
|
.. _tune-integration:
|
|
|
|
|
|
|
|
External library integrations (tune.integration)
|
|
|
|
================================================
|
|
|
|
|
|
|
|
.. contents::
|
|
|
|
:local:
|
|
|
|
:depth: 1
|
|
|
|
|
2020-10-01 19:59:23 +01:00
|
|
|
|
|
|
|
.. _tune-integration-docker:
|
|
|
|
|
|
|
|
Docker (tune.integration.docker)
|
|
|
|
--------------------------------
|
|
|
|
|
|
|
|
.. autofunction:: ray.tune.integration.docker.DockerSyncer
|
|
|
|
|
|
|
|
|
2020-09-03 18:16:08 +01:00
|
|
|
.. _tune-integration-keras:
|
|
|
|
|
|
|
|
Keras (tune.integration.keras)
|
|
|
|
------------------------------------------------------
|
|
|
|
|
|
|
|
.. autoclass:: ray.tune.integration.keras.TuneReportCallback
|
|
|
|
|
|
|
|
.. autoclass:: ray.tune.integration.keras.TuneReportCheckpointCallback
|
|
|
|
|
2020-08-31 23:30:48 +01:00
|
|
|
.. _tune-integration-kubernetes:
|
|
|
|
|
|
|
|
Kubernetes (tune.integration.kubernetes)
|
|
|
|
----------------------------------------
|
|
|
|
|
|
|
|
.. autofunction:: ray.tune.integration.kubernetes.NamespacedKubernetesSyncer
|
|
|
|
|
2021-01-19 11:55:13 -08:00
|
|
|
.. _tune-integration-mlflow:
|
|
|
|
|
|
|
|
MLflow (tune.integration.mlflow)
|
|
|
|
--------------------------------
|
|
|
|
|
2022-02-07 16:47:03 +01:00
|
|
|
:ref:`See also here <tune-mlflow-ref>`.
|
2021-01-19 11:55:13 -08:00
|
|
|
|
|
|
|
.. autoclass:: ray.tune.integration.mlflow.MLflowLoggerCallback
|
|
|
|
|
|
|
|
.. autofunction:: ray.tune.integration.mlflow.mlflow_mixin
|
|
|
|
|
|
|
|
|
2020-09-04 02:06:44 +01:00
|
|
|
.. _tune-integration-mxnet:
|
|
|
|
|
|
|
|
MXNet (tune.integration.mxnet)
|
|
|
|
------------------------------
|
|
|
|
|
|
|
|
.. autoclass:: ray.tune.integration.mxnet.TuneReportCallback
|
|
|
|
|
|
|
|
.. autoclass:: ray.tune.integration.mxnet.TuneCheckpointCallback
|
|
|
|
|
|
|
|
|
2020-08-31 23:30:48 +01:00
|
|
|
.. _tune-integration-pytorch-lightning:
|
|
|
|
|
|
|
|
PyTorch Lightning (tune.integration.pytorch_lightning)
|
|
|
|
------------------------------------------------------
|
|
|
|
|
|
|
|
.. autoclass:: ray.tune.integration.pytorch_lightning.TuneReportCallback
|
|
|
|
|
|
|
|
.. autoclass:: ray.tune.integration.pytorch_lightning.TuneReportCheckpointCallback
|
|
|
|
|
|
|
|
.. _tune-integration-torch:
|
|
|
|
|
|
|
|
Torch (tune.integration.torch)
|
|
|
|
------------------------------
|
|
|
|
|
|
|
|
.. autofunction:: ray.tune.integration.torch.DistributedTrainableCreator
|
|
|
|
|
|
|
|
.. autofunction:: ray.tune.integration.torch.distributed_checkpoint_dir
|
|
|
|
|
|
|
|
.. autofunction:: ray.tune.integration.torch.is_distributed_trainable
|
|
|
|
|
2020-09-03 16:53:35 -07:00
|
|
|
|
|
|
|
.. _tune-integration-horovod:
|
|
|
|
|
|
|
|
Horovod (tune.integration.horovod)
|
|
|
|
----------------------------------
|
|
|
|
|
|
|
|
.. autofunction:: ray.tune.integration.horovod.DistributedTrainableCreator
|
|
|
|
|
2020-08-31 23:30:48 +01:00
|
|
|
.. _tune-integration-wandb:
|
|
|
|
|
|
|
|
Weights and Biases (tune.integration.wandb)
|
|
|
|
-------------------------------------------
|
|
|
|
|
2022-02-07 16:47:03 +01:00
|
|
|
:ref:`See also here <tune-wandb-ref>`.
|
2020-08-31 23:30:48 +01:00
|
|
|
|
2020-11-17 00:08:37 +01:00
|
|
|
.. autoclass:: ray.tune.integration.wandb.WandbLoggerCallback
|
2020-08-31 23:30:48 +01:00
|
|
|
|
2020-09-02 19:16:09 +01:00
|
|
|
.. autofunction:: ray.tune.integration.wandb.wandb_mixin
|
|
|
|
|
|
|
|
|
|
|
|
.. _tune-integration-xgboost:
|
|
|
|
|
|
|
|
XGBoost (tune.integration.xgboost)
|
|
|
|
----------------------------------
|
|
|
|
|
|
|
|
.. autoclass:: ray.tune.integration.xgboost.TuneReportCallback
|
|
|
|
|
|
|
|
.. autoclass:: ray.tune.integration.xgboost.TuneReportCheckpointCallback
|
2021-07-24 06:21:13 +02:00
|
|
|
|
|
|
|
|
|
|
|
.. _tune-integration-lightgbm:
|
|
|
|
|
|
|
|
LightGBM (tune.integration.lightgbm)
|
|
|
|
------------------------------------
|
|
|
|
|
|
|
|
.. autoclass:: ray.tune.integration.lightgbm.TuneReportCallback
|
|
|
|
|
|
|
|
.. autoclass:: ray.tune.integration.lightgbm.TuneReportCheckpointCallback
|