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

[raysgd] Better user errors! (#7546)

Browse Source 
* format

* callable

* Update python/ray/util/sgd/torch/torch_trainer.py

Co-Authored-By: Edward Oakes <ed.nmi.oakes@gmail.com>

* Update python/ray/util/sgd/torch/torch_trainer.py

Co-Authored-By: Edward Oakes <ed.nmi.oakes@gmail.com>

* data

* torchtrainer

* num_rep

Co-authored-by: Edward Oakes <ed.nmi.oakes@gmail.com>
This commit is contained in:
Richard Liaw
2020-03-10 18:58:19 -07:00
committed by GitHub
co-authored by Edward Oakes
parent 7b609ca211
commit 6163b21458
2 changed files with 51 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/util/sgd/torch/torch_runner.py
+6
View File
@@ -95,6 +95,12 @@ class TorchRunner:
with FileLock(os.path.join(tempfile.gettempdir(), ".ray_data.lock")):
loaders = self.data_creator(self.config)
train_loader, val_loader = self._validate_loaders(loaders)
if not isinstance(train_loader, torch.utils.data.DataLoader):
logger.warning(
"TorchTrainer data_creator return values are no longer "
"wrapped as DataLoaders. Users must return DataLoader(s) "
"in data_creator. This warning will be removed in "
"a future version of Ray.")
self.train_loader, self.validation_loader = train_loader, val_loader