wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-19 11:27:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add ray.util package and move libraries from experimental (#7100)

Browse Source
This commit is contained in:
Eric Liang
2020-02-18 13:43:19 -08:00
committed by GitHub
parent fae99ecb8e
commit 5df801605e
113 changed files with 305 additions and 637 deletions
Download Patch File Download Diff File Expand all files Collapse all files
doc/examples/doc_code/raysgd_torch_signatures.py
+2 -2
View File
@@ -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,