Toggle navigation sidebar
Toggle in-page Table of Contents
Ray 3.0.0.dev0
Welcome to Ray!
Ray
Overview
Getting Started
Installation
Use Cases
Example Gallery
Ecosystem
Ray Core
Ray Data
Ray Train
Key Concepts
Distributed PyTorch
More Frameworks
Ray Train Internals
Examples
Ray Train FAQ
Ray Train API
ray.train.trainer.BaseTrainer
ray.train.data_parallel_trainer.DataParallelTrainer
ray.train.DataConfig
ray.train.gbdt_trainer.GBDTTrainer
ray.train.trainer.BaseTrainer.fit
ray.train.trainer.BaseTrainer.setup
ray.train.trainer.BaseTrainer.preprocess_datasets
ray.train.trainer.BaseTrainer.training_loop
ray.train.trainer.BaseTrainer.as_trainable
ray.train.backend.Backend
ray.train.backend.BackendConfig
ray.train.context.TrainContext
ray.train.get_context
ray.train.get_dataset_shard
ray.train.report
ray.train.Checkpoint
ray.train.get_context
ray.train.context.TrainContext
ray.train.torch.TorchTrainer
ray.train.torch.TorchConfig
ray.train.torch.TorchCheckpoint
ray.train.torch.prepare_model
ray.train.torch.prepare_optimizer
ray.train.torch.prepare_data_loader
ray.train.torch.get_device
ray.train.torch.accelerate
ray.train.torch.backward
ray.train.torch.enable_reproducibility
ray.train.lightning.LightningTrainer
ray.train.lightning.LightningConfigBuilder
ray.train.lightning.LightningCheckpoint
ray.train.lightning.LightningPredictor
ray.train.lightning.LightningPredictor.call_model
ray.train.lightning.LightningPredictor.from_checkpoint
ray.train.lightning.LightningPredictor.from_pandas_udf
ray.train.lightning.LightningPredictor.get_preprocessor
ray.train.lightning.LightningPredictor.predict
ray.train.lightning.LightningPredictor.preferred_batch_format
ray.train.lightning.LightningPredictor.set_preprocessor
ray.train.tensorflow.TensorflowTrainer
ray.train.tensorflow.TensorflowConfig
ray.train.tensorflow.TensorflowCheckpoint
ray.train.tensorflow.prepare_dataset_shard
ray.train.horovod.HorovodTrainer
ray.train.horovod.HorovodConfig
ray.train.xgboost.XGBoostTrainer
ray.train.xgboost.XGBoostCheckpoint
ray.train.lightgbm.LightGBMTrainer
ray.train.lightgbm.LightGBMCheckpoint
ray.train.huggingface.TransformersTrainer
ray.train.huggingface.TransformersCheckpoint
ray.train.huggingface.AccelerateTrainer
ray.train.sklearn.SklearnTrainer
ray.train.sklearn.SklearnCheckpoint
ray.train.mosaic.MosaicTrainer
ray.train.trainer.BaseTrainer.restore
ray.train.data_parallel_trainer.DataParallelTrainer.restore
Ray Tune
Ray Serve
Ray RLlib
More Libraries
Ray Clusters
Monitoring and Debugging
References
Developer Guides
repository
open issue
suggest edit
.rst
.pdf
ray.train.lightning.LightningPredictor.preferred_batch_format
ray.train.lightning.LightningPredictor.preferred_batch_format
#
classmethod
LightningPredictor.
preferred_batch_format
(
)
→
ray.air.util.data_batch_conversion.BatchFormat
#
DeveloperAPI:
This API may change across minor Ray releases.