mirror of
https://github.com/vale981/ray
synced 2025-03-08 19:41:38 -05:00
759966781f
Co-authored-by: Xiaowei Jiang <xwjiang2010@gmail.com> Co-authored-by: matthewdeng <matthew.j.deng@gmail.com> Co-authored-by: Kai Fricke <krfricke@users.noreply.github.com>
72 lines
2 KiB
Python
72 lines
2 KiB
Python
# flake8: noqa
|
|
# isort: skip_file
|
|
|
|
|
|
# __config_scaling_1__
|
|
from ray.air import session
|
|
from ray.train.torch import TorchTrainer
|
|
from ray.air.config import ScalingConfig
|
|
import ray.data
|
|
|
|
|
|
def train_loop_per_worker():
|
|
# By default, bulk loading is used and returns a Dataset object.
|
|
data_shard = session.get_dataset_shard("train")
|
|
|
|
# Manually iterate over the data 10 times (10 epochs).
|
|
for _ in range(10):
|
|
for batch in data_shard.iter_batches():
|
|
print("Do some training on batch", batch)
|
|
|
|
|
|
trainer = TorchTrainer(
|
|
train_loop_per_worker,
|
|
scaling_config=ScalingConfig(num_workers=1),
|
|
datasets={"train": ray.data.range_tensor(1000)},
|
|
)
|
|
trainer.fit()
|
|
# __config_scaling_1_end__
|
|
|
|
# __config_scaling_2__
|
|
from ray.air import session
|
|
from ray.train.torch import TorchTrainer
|
|
import ray.data
|
|
from ray.air.config import ScalingConfig
|
|
from ray import tune
|
|
from ray.tune.tuner import Tuner
|
|
from ray.tune.tune_config import TuneConfig
|
|
|
|
|
|
def train_loop_per_worker():
|
|
# By default, bulk loading is used and returns a Dataset object.
|
|
data_shard = session.get_dataset_shard("train")
|
|
|
|
# Manually iterate over the data 10 times (10 epochs).
|
|
for _ in range(10):
|
|
for batch in data_shard.iter_batches():
|
|
print("Do some training on batch", batch)
|
|
|
|
|
|
trainer = TorchTrainer(
|
|
train_loop_per_worker,
|
|
scaling_config=ScalingConfig(num_workers=1),
|
|
datasets={"train": ray.data.range_tensor(1000)},
|
|
)
|
|
param_space = {
|
|
"scaling_config": ScalingConfig(num_workers=tune.grid_search([1, 2])),
|
|
"params": {
|
|
"objective": "binary:logistic",
|
|
"tree_method": "approx",
|
|
"eval_metric": ["logloss", "error"],
|
|
"eta": tune.loguniform(1e-4, 1e-1),
|
|
"subsample": tune.uniform(0.5, 1.0),
|
|
"max_depth": tune.randint(1, 9),
|
|
},
|
|
}
|
|
tuner = Tuner(
|
|
trainable=trainer,
|
|
param_space=param_space,
|
|
tune_config=TuneConfig(mode="min", metric="train-error"),
|
|
)
|
|
results = tuner.fit()
|
|
# __config_scaling_2_end__
|