mirror of
https://github.com/wassname/ray.git
synced 2026-10-29 13:30:42 +08:00
Add ray.util package and move libraries from experimental (#7100)
This commit is contained in:
1 parent
fae99ecb8e
commit
5df801605e
113 files changed
+305
-637
No files matched your search
@@ -46,7 +46,7 @@ def optimizer_creator(model, config):
|
||||
|
||||
|
||||
# __torch_data_start__
|
||||
from ray.experimental.sgd.pytorch.examples.train_example import LinearDataset
|
||||
from ray.util.sgd.pytorch.examples.train_example import LinearDataset
|
||||
|
||||
def data_creator(config):
|
||||
"""Constructs torch.utils.data.Dataset objects.
|
||||
@@ -108,7 +108,7 @@ ray.init()
|
||||
# __torch_ray_end__
|
||||
|
||||
# __torch_trainer_start__
|
||||
from ray.experimental.sgd import PyTorchTrainer
|
||||
from ray.util.sgd import PyTorchTrainer
|
||||
|
||||
trainer = PyTorchTrainer(
|
||||
model_creator,
|
||||
|
||||
Reference in new issue
Block a user