mirror of
https://github.com/vale981/ray
synced 2025-03-06 18:41:40 -05:00
![]() This PR adds .iter_torch_batches() and .iter_tf_batches() convenience APIs, which takes care of ML framework tensor conversion, the narrow tensor waste for the .iter_batches() call ("numpy" format), and unifies batch formats around two options: a single tensor for simple/pure-tensor/single-column datasets, and a dictionary of tensors for multi-column datasets. |
||
---|---|---|
.. | ||
air_ingest.py | ||
air_key_concepts.py | ||
checkpoint_mlflow.py | ||
checkpoint_usage.py | ||
config_scaling.py | ||
output.txt | ||
predictors.py | ||
preprocessors.py | ||
pytorch_starter.py | ||
tf_starter.py |